Searched defs:original_shape (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| 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
|
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | nearest_upsample.py | 33 def new_shape_for_upsample(original_shape, scales, axis): argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | mirror_pad_op.cc | 32 const xla::Shape& original_shape, in DoMirrorPad()
119 const xla::Shape& original_shape, in DoMirrorPadGrad()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | layout_util.cc | 67 xla::XlaBuilder* builder, xla::XlaOp original, xla::Shape original_shape, in ReshapeWithCorrectRepresentationAndSharding()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | layout_util.cc | 83 xla::XlaBuilder* builder, xla::XlaOp original, xla::Shape original_shape, in ReshapeWithCorrectRepresentationAndSharding()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_normalization.cc | 172 auto original_shape = hlo->shape(); in ChangeOutputTypeThenInsertConvertBack() local
348 auto original_shape = hlo->shape(); in HandleMultipleOutputs() local
|
| 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
|
| 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()
|
| /external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/ |
| D | sparse_reshape_op_test.py | 354 def _MakeAndReshapeTensor(self, tensor_class, original_shape, target_shape): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | strided_slice_op.cc | 355 TensorShape original_shape = old_lhs->shape(); in Compute() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | prng.cc | 222 const Shape& original_shape) { in CombineShapePair()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| 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
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 323 TensorShapeProto MaybeGetMinimumShape(const TensorShapeProto& original_shape, in MaybeGetMinimumShape()
|