Searched defs:original_shape (Results 1 – 9 of 9) sorted by relevance
54 std::unique_ptr<ProgramShape> original_shape = in TEST_F() local87 std::unique_ptr<ProgramShape> original_shape = in XLA_TEST_F() local134 std::unique_ptr<ProgramShape> original_shape = in TEST_F() local
147 tensorflow::TensorShape original_shape; in __anon96b295990202() local
34 Output GetVectorizedShape(Scope* s, Output tensor, Output original_shape) { in GetVectorizedShape()
30 const xla::Shape& original_shape, in DoMirrorPad()
269 const tensorflow::TensorShape& original_shape, in do_periodic_resample_op()368 const tensorflow::TensorShape& original_shape, in create_grad_tensor()
373 def calculate_reshape(original_shape, new_shape, validate=False, name=None): argument
237 auto original_shape = hlo->shape(); in HandleMultipleOutputs() local
333 TensorShape original_shape = old_lhs->shape(); in Compute() local
183 TensorShapeProto MaybeGetMinimumShape(const TensorShapeProto& original_shape, in MaybeGetMinimumShape()