Searched refs:partial_tensor_shape (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | type_to_shape.cc | 85 tensorflow::PartialTensorShape partial_tensor_shape = in TypeToShape() local
89 if (!partial_tensor_shape.AsTensorShape(&fully_defined_tensor_shape)) { in TypeToShape()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_tensor_to_tensor_op.cc | 156 PartialTensorShape partial_tensor_shape; in CalculateOutputSize() local
158 &partial_tensor_shape)); in CalculateOutputSize()
159 partial_tensor_shape.AsProto(&shape_proto); in CalculateOutputSize()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | BUILD | 75 "partial_tensor_shape.h",
216 "partial_tensor_shape.h",
402 "partial_tensor_shape.h",
671 "partial_tensor_shape.h",
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 778 const PartialTensorShape& partial_tensor_shape) { in GetPadding() argument
780 if (partial_tensor_shape.dim_size(split_dim) <= 0) { in GetPadding()
784 partial_tensor_shape.dim_size(split_dim), num_splits); in GetPadding()
786 per_split_size * num_splits - partial_tensor_shape.dim_size(split_dim); in GetPadding()
793 const PartialTensorShape& partial_tensor_shape, int replica_id, in CreateOrGetSplitNodesForInputSharding() argument
837 num_split, split_dimension, partial_tensor_shape.dims(), in CreateOrGetSplitNodesForInputSharding()
838 GetPadding(split_dimension, num_split, partial_tensor_shape), in CreateOrGetSplitNodesForInputSharding()
861 num_split, split_dimension, partial_tensor_shape.dims(), in CreateOrGetSplitNodesForInputSharding()
862 GetPadding(split_dimension, num_split, partial_tensor_shape), in CreateOrGetSplitNodesForInputSharding()
|
| /external/tensorflow/tensorflow/core/ |
| D | BUILD | 445 "//tensorflow/core/framework:partial_tensor_shape.h",
|
| /external/tensorflow/ |
| D | RELEASE.md | 4650 * `partial_tensor_shape.h` -> `../framework/partial_tensor_shape.h`
|