Home
last modified time | relevance | path

Searched defs:original_shape (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dreplay_test.cc54 std::unique_ptr<ProgramShape> original_shape = in TEST_F() local
87 std::unique_ptr<ProgramShape> original_shape = in XLA_TEST_F() local
134 std::unique_ptr<ProgramShape> original_shape = in TEST_F() local
/external/tensorflow/tensorflow/lite/testing/op_tests/
Dnearest_upsample.py37 def new_shape_for_upsample(original_shape, scales, axis): argument
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dmirror_pad_op.cc32 const xla::Shape& original_shape, in DoMirrorPad()
119 const xla::XlaOp t, const xla::Shape& original_shape, in DoMirrorPadGrad()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/
Dreshape_vectorizer.cc34 Output GetVectorizedShape(Scope* s, Output tensor, Output original_shape) { in GetVectorizedShape()
/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_normalization.cc173 auto original_shape = hlo->shape(); in ChangeOutputTypeThenInsertConvertBack() local
349 auto original_shape = hlo->shape(); in HandleMultipleOutputs() local
Dhlo_rematerialization.cc1246 const Shape& original_shape = hlo->shape(); in GetCompactShape() local
1415 const Shape& original_shape = item->instruction->shape(); in PickRematerializationCandidates() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dsparse_reshape_op_test.py336 def _MakeAndReshapeTensor(self, tensor_class, original_shape, target_shape): argument
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_helpers.cc181 xla::XlaBuilder* builder, xla::XlaOp original, xla::Shape original_shape, in ReshapeWithCorrectRepresentationAndSharding()
/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op.cc350 TensorShape original_shape = old_lhs->shape(); in Compute() local
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dprng.cc221 const Shape& original_shape) { in CombineShapePair()
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc322 TensorShapeProto MaybeGetMinimumShape(const TensorShapeProto& original_shape, in MaybeGetMinimumShape()