Home
last modified time | relevance | path

Searched defs:partial_tensor_shape (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/
Dtype_to_shape.cc85 tensorflow::PartialTensorShape partial_tensor_shape = in TypeToShape() local
/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_to_tensor_op.cc156 PartialTensorShape partial_tensor_shape; in CalculateOutputSize() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc778 const PartialTensorShape& partial_tensor_shape) { in GetPadding()
793 const PartialTensorShape& partial_tensor_shape, int replica_id, in CreateOrGetSplitNodesForInputSharding()