Searched defs:partial_tensor_shape (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | distributed_tpu_rewrite_pass.cc | 789 const PartialTensorShape& partial_tensor_shape) { in GetPadding()
804 const PartialTensorShape& partial_tensor_shape, int replica_id, in CreateOrGetSplitNodesForInputSharding()
920 const PartialTensorShape& partial_tensor_shape, in CreateXlaSplitOp()
976 const PartialTensorShape& partial_tensor_shape, in ShardInputWithXlaSplitOp()
1009 const PartialTensorShape& partial_tensor_shape, Node* orig_src, in CreateOrGetXlaSplitNodeForShardedPerReplicaArg()
1036 const PartialTensorShape& partial_tensor_shape, Node* orig_src, in CreateOrGetXlaSplitNodeForDistributedArg()
1066 const PartialTensorShape& partial_tensor_shape, Node* orig_src, in CreateOrGetXlaSplitNodeForVariableArg()
1285 const PartialTensorShape& partial_tensor_shape, in CreateXlaConcatNode()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | ragged_tensor_to_tensor_op.cc | 156 PartialTensorShape partial_tensor_shape; in CalculateOutputSize() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | tpu_functional_ops.cc | 439 PartialTensorShape partial_tensor_shape; in GetInputOutputInfo() local
2051 PartialTensorShape partial_tensor_shape; in InferShapesWithResourceVar() local
|