Searched defs:original_shape (Results 1 – 11 of 11) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | replay_test.cc | 54 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/ |
D | nearest_upsample.py | 37 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::XlaOp t, const xla::Shape& original_shape, in DoMirrorPadGrad()
|
/external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/ |
D | reshape_vectorizer.cc | 34 Output GetVectorizedShape(Scope* s, Output tensor, Output original_shape) { in GetVectorizedShape()
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | bfloat16_normalization.cc | 173 auto original_shape = hlo->shape(); in ChangeOutputTypeThenInsertConvertBack() local 349 auto original_shape = hlo->shape(); in HandleMultipleOutputs() local
|
D | hlo_rematerialization.cc | 1246 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/ |
D | sparse_reshape_op_test.py | 336 def _MakeAndReshapeTensor(self, tensor_class, original_shape, target_shape): argument
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | xla_helpers.cc | 181 xla::XlaBuilder* builder, xla::XlaOp original, xla::Shape original_shape, in ReshapeWithCorrectRepresentationAndSharding()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | strided_slice_op.cc | 350 TensorShape original_shape = old_lhs->shape(); in Compute() local
|
/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
D | prng.cc | 221 const Shape& original_shape) { in CombineShapePair()
|
/external/tensorflow/tensorflow/core/grappler/costs/ |
D | op_level_cost_estimator.cc | 322 TensorShapeProto MaybeGetMinimumShape(const TensorShapeProto& original_shape, in MaybeGetMinimumShape()
|