| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | scatter_nd_op.cc | 34 Status ValidateUpdateShape(const TensorShape& buffer_shape, in ValidateUpdateShape()
95 TensorShape buffer_shape; in Compile() local
150 TensorShape buffer_shape = context->InputShape(0); in CompileTensorScatter() local
|
| H A D | tensor_list_utils.cc | 139 Status GetTensorListBufferShape(xla::XlaOp list, xla::Shape* buffer_shape) { in GetTensorListBufferShape()
213 auto buffer_shape = xla::ShapeUtil::GetTupleElementShape(list_shape, 0); in GetLeadingDimForTensorList() local
517 const xla::Shape& buffer_shape = in ExecuteTensorListGetItem() local
|
| H A D | segment_reduction_ops.cc | 83 TensorShape buffer_shape = data_shape; in Compile() local
|
| H A D | tensor_list_ops.cc | 419 TensorShape buffer_shape; in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| H A D | segment_reduction_ops.cc | 73 TensorShape buffer_shape = data_shape; in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | gpu_transfer_manager.cc | 102 const Shape& buffer_shape = in ReadDynamicShapes() local
|
| H A D | buffer_comparator.cc | 613 const Shape& buffer_shape, in DeviceCompare()
|
| H A D | gpu_conv_algorithm_picker.cc | 656 const Shape& buffer_shape) { in PickBestAlgorithmNoCacheCuda()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorIteratorReduce.cpp | 46 auto buffer_shape = DimVector(unsqueezed.sizes()); in two_pass_reduction() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | cpu_xfeed.cc | 283 const Shape& buffer_shape = in ReadDynamicShapesOnCpu() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | collection_ops_util.cc | 153 llvm::SmallVector<int64_t, 8> buffer_shape; in CreateInitBufferValue() local
|
| H A D | tensor_array_ops_decomposition.cc | 335 llvm::SmallVector<int64_t, 8> buffer_shape; in HandleTensorArraySplitV3Op() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | transfer_manager.cc | 208 const Shape& buffer_shape = in ReadDynamicShapes() local
|
| H A D | layout_assignment.cc | 649 const Shape& buffer_shape = instruction->operand(0)->shape(); in AddMandatoryConstraints() local
|
| H A D | hlo_verifier.cc | 523 Status CheckBufferOffset(const Shape& buffer_shape, in CheckBufferOffset()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | xla_compiler.cc | 939 xla::Shape buffer_shape; in XLAShapeForArgument() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | pjrt_stream_executor_client.cc | 1645 const Shape& buffer_shape, in CheckCompatibleShapes()
|