| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | replay_test.cc | 52 std::unique_ptr<ProgramShape> original_shape = in TEST_F() local
84 std::unique_ptr<ProgramShape> original_shape = in XLA_TEST_F() local
128 std::unique_ptr<ProgramShape> original_shape = in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | mirror_pad_op.cc | 32 const xla::Shape& original_shape, in DoMirrorPad()
119 const xla::Shape& original_shape, in DoMirrorPadGrad()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | bfloat16_normalization.cc | 172 auto original_shape = hlo->shape(); in ChangeOutputTypeThenInsertConvertBack() local
348 auto original_shape = hlo->shape(); in HandleMultipleOutputs() local
|
| H A D | hlo_rematerialization.cc | 1257 const Shape& original_shape = hlo->shape(); in GetCompactShape() local
1430 const Shape& original_shape = item->instruction->shape(); in PickRematerializationCandidates() local
|
| H A D | algebraic_simplifier.cc | 1147 const Shape& original_shape, const Shape& result_shape, in ReshapeLayoutDimensions()
1229 const Shape& original_shape, const Shape& bitcast_shape, in InvertBitcastDimMap()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | LinearAlgebraUtils.h | 550 inline Tensor get_linear_indices(int64_t numel, IntArrayRef original_shape, IntArrayRef broadcast_s… in get_linear_indices()
563 IntArrayRef original_shape, in BroadcastLinearIndices()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/testing/op_tests/ |
| H A D | nearest_upsample.py | 33 def new_shape_for_upsample(original_shape, scales, axis): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ |
| H A D | layout_util.cc | 67 xla::XlaBuilder* builder, xla::XlaOp original, xla::Shape original_shape, in ReshapeWithCorrectRepresentationAndSharding()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | layout_util.cc | 83 xla::XlaBuilder* builder, xla::XlaOp original, xla::Shape original_shape, in ReshapeWithCorrectRepresentationAndSharding()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/ |
| H A D | sparse_reshape_op_test.py | 354 def _MakeAndReshapeTensor(self, tensor_class, original_shape, target_shape): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | strided_slice_op.cc | 355 TensorShape original_shape = old_lhs->shape(); in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | prng.cc | 222 const Shape& original_shape) { in CombineShapePair()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | TensorIterator.cpp | 1273 IntArrayRef original_shape = config.static_shape_ ? shape_ : op.tensor_base().sizes(); in compute_strides() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | spmd_partitioner.cc | 1730 Shape original_shape = to_reshape.hlo()->shape(); in SplitReshapeHelper() local
1753 Shape original_shape = to_reshape.hlo()->shape(); in MergeReshapeHelper() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | op_level_cost_estimator.cc | 323 TensorShapeProto MaybeGetMinimumShape(const TensorShapeProto& original_shape, in MaybeGetMinimumShape()
|