-
Notifications
You must be signed in to change notification settings - Fork 18
Open
Labels
enhancementNew feature or requestNew feature or requestjavaPull requests that update Java codePull requests that update Java codetensor shapes
Milestone
Description
The tf.Tensor(op, value_index, dtype) constructor allows creating a tensor handle from an existing operation output. To support shape inference for such tensors, we need to:
- Identify the operation (or generator) that the
opargument corresponds to. - Delegate the shape computation to that generator, using the
value_indexto select the correct output.
This requires the ability to trace the op argument back to its creating logic (e.g., a tf.matmul call) and instantiate the appropriate TensorGenerator.
This task is blocked by #340, which addresses the general mechanism for operations involving calls to other tensor generators.
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
enhancementNew feature or requestNew feature or requestjavaPull requests that update Java codePull requests that update Java codetensor shapes