Searched defs:partial_tensor_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | type_to_shape.cc | 85 tensorflow::PartialTensorShape partial_tensor_shape = in TypeToShape() local
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 791 const PartialTensorShape& partial_tensor_shape) { in GetPadding()
806 const PartialTensorShape& partial_tensor_shape, int replica_id, in CreateOrGetSplitNodesForInputSharding()
922 const PartialTensorShape& partial_tensor_shape, in CreateXlaSplitOp()
980 const PartialTensorShape& partial_tensor_shape, in ShardInputWithXlaSplitOp()
1013 const PartialTensorShape& partial_tensor_shape, Node* orig_src, in CreateOrGetXlaSplitNodeForShardedPerReplicaArg()
1040 const PartialTensorShape& partial_tensor_shape, Node* orig_src, in CreateOrGetXlaSplitNodeForDistributedArg()
1070 const PartialTensorShape& partial_tensor_shape, Node* orig_src, in CreateOrGetXlaSplitNodeForVariableArg()
1295 const PartialTensorShape& partial_tensor_shape, in CreateXlaConcatNode()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_tensor_to_tensor_op.cc | 156 PartialTensorShape partial_tensor_shape; in CalculateOutputSize() local
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_functional_ops.cc | 423 PartialTensorShape partial_tensor_shape; in GetInputOutputInfo() local
1966 PartialTensorShape partial_tensor_shape; in InferShapesWithResourceVar() local
|