| /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() argument
791 if (partial_tensor_shape.dim_size(split_dim) <= 0) { in GetPadding()
795 partial_tensor_shape.dim_size(split_dim), num_splits); in GetPadding()
797 per_split_size * num_splits - partial_tensor_shape.dim_size(split_dim); in GetPadding()
804 const PartialTensorShape& partial_tensor_shape, int replica_id, in CreateOrGetSplitNodesForInputSharding() argument
848 num_split, split_dimension, partial_tensor_shape.dims(), in CreateOrGetSplitNodesForInputSharding()
849 GetPadding(split_dimension, num_split, partial_tensor_shape), in CreateOrGetSplitNodesForInputSharding()
872 num_split, split_dimension, partial_tensor_shape.dims(), in CreateOrGetSplitNodesForInputSharding()
873 GetPadding(split_dimension, num_split, partial_tensor_shape), in CreateOrGetSplitNodesForInputSharding()
920 const PartialTensorShape& partial_tensor_shape, in CreateXlaSplitOp() argument
[all …]
|
| /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
158 &partial_tensor_shape)); in CalculateOutputSize()
159 partial_tensor_shape.AsProto(&shape_proto); in CalculateOutputSize()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | tpu_functional_ops.cc | 439 PartialTensorShape partial_tensor_shape; in GetInputOutputInfo() local
441 input_shapes.data(), input_shapes.size(), &partial_tensor_shape); in GetInputOutputInfo()
442 InferredShape inferred_shape = {partial_tensor_shape}; in GetInputOutputInfo()
2051 PartialTensorShape partial_tensor_shape; in InferShapesWithResourceVar() local
2054 &partial_tensor_shape); in InferShapesWithResourceVar()
2055 InferredShape inferred_shape = {partial_tensor_shape}; in InferShapesWithResourceVar()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | BUILD | 73 "partial_tensor_shape.h",
218 "partial_tensor_shape.h",
413 "partial_tensor_shape.h",
687 "partial_tensor_shape.h",
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ |
| H A D | BUILD | 468 "//tensorflow/core/framework:partial_tensor_shape.h",
|
| /aosp_15_r20/out/soong/.intermediates/external/tensorflow/libtflite_flex_delegate/android_arm64_armv8-2a_cortex-a55_static_apex33/obj/external/tensorflow/tensorflow/core/kernels/ |
| D | inplace_ops.o.d | 1311 external/tensorflow/tensorflow/core/framework/partial_tensor_shape.h \
|
| D | strided_slice_op_inst_2.o.d | 1314 external/tensorflow/tensorflow/core/framework/partial_tensor_shape.h \
|
| D | strided_slice_op_inst_7.o.d | 1314 external/tensorflow/tensorflow/core/framework/partial_tensor_shape.h \
|
| D | concat_op.o.d | 1303 external/tensorflow/tensorflow/core/framework/partial_tensor_shape.h \
|
| D | split_op.o.d | 1313 external/tensorflow/tensorflow/core/framework/partial_tensor_shape.h \
|
| D | strided_slice_op_inst_5.o.d | 1314 external/tensorflow/tensorflow/core/framework/partial_tensor_shape.h \
|
| D | no_op.o.d | 1302 external/tensorflow/tensorflow/core/framework/partial_tensor_shape.h \
|
| D | strided_slice_op_inst_3.o.d | 1314 external/tensorflow/tensorflow/core/framework/partial_tensor_shape.h \
|
| D | tensor_to_hash_bucket_op.o.d | 1302 external/tensorflow/tensorflow/core/framework/partial_tensor_shape.h \
|
| D | bcast_ops.o.d | 1278 external/tensorflow/tensorflow/core/framework/partial_tensor_shape.h \
|
| D | shape_ops.o.d | 1305 external/tensorflow/tensorflow/core/framework/partial_tensor_shape.h \
|
| D | save_restore_v2_ops.o.d | 1305 external/tensorflow/tensorflow/core/framework/partial_tensor_shape.h \
|
| D | strided_slice_op_inst_4.o.d | 1314 external/tensorflow/tensorflow/core/framework/partial_tensor_shape.h \
|
| /aosp_15_r20/out/soong/.intermediates/external/tensorflow/libtflite_flex_delegate/android_arm64_armv8-2a_cortex-a55_static_apex33/obj/external/tensorflow/tensorflow/core/ops/ |
| D | string_ops.o.d | 814 external/tensorflow/tensorflow/core/framework/partial_tensor_shape.h \
|
| D | dataset_ops.o.d | 803 external/tensorflow/tensorflow/core/framework/partial_tensor_shape.h \
|
| D | nn_ops.o.d | 803 external/tensorflow/tensorflow/core/framework/partial_tensor_shape.h \
|
| D | io_ops.o.d | 803 external/tensorflow/tensorflow/core/framework/partial_tensor_shape.h \
|
| /aosp_15_r20/out/soong/.intermediates/external/tensorflow/libtflite_flex_delegate/android_arm64_armv8-2a_cortex-a55_static_apex33/obj/external/tensorflow/tensorflow/core/common_runtime/ |
| D | propagator_debug_utils.o.d | 1304 external/tensorflow/tensorflow/core/framework/partial_tensor_shape.h \
|
| /aosp_15_r20/out/soong/.intermediates/external/tensorflow/libtflite_flex_delegate/android_arm64_armv8-2a_cortex-a55_static_apex33/obj/external/tensorflow/tensorflow/core/util/ |
| D | guarded_philox_random.o.d | 1302 external/tensorflow/tensorflow/core/framework/partial_tensor_shape.h \
|
| /aosp_15_r20/out/soong/.intermediates/external/tensorflow/libtflite_flex_delegate/android_arm64_armv8-2a_cortex-a55_static_apex33/obj/external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.o.d | 802 external/tensorflow/tensorflow/core/framework/partial_tensor_shape.h \
|