Home
last modified time | relevance | path

Searched defs:reshape (Results 1 – 25 of 68) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dcanonicalize_all_gather_for_cse_test.cc66 const HloInstruction* const reshape = in TEST_F() local
86 const HloInstruction* const reshape = in TEST_F() local
106 const HloInstruction* const reshape = in TEST_F() local
126 const HloInstruction* const reshape = in TEST_F() local
146 const HloInstruction* const reshape = in TEST_F() local
166 const HloInstruction* const reshape = in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dspace_to_batch_converter_test.cc56 auto reshape = root->operand(0)->operand(0); in TEST_F() local
105 auto reshape = root->operand(0)->operand(0); in TEST_F() local
139 auto reshape = root->operand(0)->operand(0); in TEST_F() local
Dhlo_computation_test.cc548 HloInstruction* reshape = in TEST_F() local
583 HloInstruction* reshape = in TEST_F() local
619 HloInstruction* reshape = in TEST_F() local
Dhlo_instruction_test.cc1208 HloInstruction* reshape = in TEST_F() local
1307 HloInstruction* reshape = in TEST_F() local
1439 HloInstruction* reshape = in TEST_F() local
1618 HloInstruction* reshape = in TEST_F() local
1660 HloInstruction* reshape = in TEST_F() local
1716 HloInstruction* reshape = in TEST_F() local
Ddynamic_padder.cc341 HloInstruction* reshape, int64 input_dim, in RewriteDynamicReshapeSplitInput()
556 HloInstruction* reshape, absl::Span<const int64> input_dims, in RewriteDynamicReshapeCombineInput()
705 HloInstruction* reshape, absl::Span<const int64> input_dims, in RewriteDynamicReshapeSingleGroup()
1389 HloInstruction* reshape, in RewriteDynamicReshape()
Ddynamic_dimension_inference_test.cc574 auto* reshape = builder.AddInstruction( in TEST_F() local
609 auto* reshape = builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F() local
635 auto* reshape = builder.AddInstruction( in TEST_F() local
/external/tensorflow/tensorflow/lite/experimental/examples/lstm/
Dinput_data.py131 reshape=True, argument
271 reshape=True, argument
/external/tensorflow/tensorflow/compiler/jit/
Dpartially_decluster_pass_test.cc308 Output reshape = ops::Reshape(s.WithOpName("reshape"), reshape_input, shape); in TEST() local
335 Output reshape = ops::Reshape(s.WithOpName("reshape"), reshape_input, shape); in TEST() local
360 Output reshape = ops::Reshape(s.WithOpName("reshape"), reshape_input, shape); in TEST() local
385 Output reshape = ops::Reshape(s.WithOpName("reshape"), reshape_input, shape); in TEST() local
/external/tensorflow/tensorflow/python/training/
Dsaver.py317 reshape, argument
364 restore_sequentially, reshape): argument
421 reshape=False, argument
474 reshape=False, argument
691 reshape=False, argument
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dreduction_degenerate_dim_remover.cc69 std::unique_ptr<HloInstruction> reshape = in HandleReduce() local
/external/tensorflow/tensorflow/core/tpu/kernels/xla/
Dget_item_op.cc67 auto reshape = xla::Reshape(slice, dims, out_sizes); in Compile() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dconst_analysis_test.cc157 Output reshape = ops::Reshape(root, arg1, add); in TEST() local
181 Output reshape = ops::Reshape(root, arg1, add); in TEST() local
/external/tensorflow/tensorflow/lite/micro/kernels/
Dreshape.cc28 namespace reshape { namespace
/external/tensorflow/tensorflow/core/util/
Dbcast.h35 const gtl::InlinedVector<int64, 4>& reshape, in ComputeBatchIndices()
94 const Vec& reshape(int i) const { return reshape_[i]; } in reshape() function
/external/webrtc/sdk/objc/components/renderer/opengl/
DRTCNSGLVideoView.m80 - (void)reshape { method in RTCNSGLVideoView
/external/tensorflow/tensorflow/core/kernels/
Dunravel_index_op.cc110 Eigen::array<Eigen::Index, 2> reshape{ in Compute() local
Dscoped_allocator_ops_test.cc145 bool reshape, const string& name, int32 id, in BuildNodeDefWithReshape()
162 void MakeOp(const TensorShape& shape, DataType dtype, bool reshape, in MakeOp()
/external/llvm-project/clang/test/Analysis/
Dmalloc-interprocedural.c77 static char *reshape(char *in) { in reshape() function
/external/clang/test/Analysis/
Dmalloc-interprocedural.c77 static char *reshape(char *in) { in reshape() function
/external/tensorflow/tensorflow/lite/kernels/
Dreshape.cc28 namespace reshape { namespace
/external/tensorflow/tensorflow/python/kernel_tests/
Dreshape_op_test.py125 def reshape(x): function
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorBase.h738 reshape(const NewDimensions& newDimensions) const { in reshape() function
907 reshape(const NewDimensions& newDimensions) const { in reshape() function
912 reshape(const NewDimensions& newDimensions) { in reshape() function
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbitcast_convert_test.cc134 auto reshape = Reshape(input, /*dimensions=*/{0}, /*new_sizes=*/{}); in TEST_F() local
Dslice_test.cc174 auto reshape = Reshape(original, {24, 3, 2, 7}); in TEST_F() local
184 auto reshape = Reshape(original, {2 * 3 * 5, 7}); in TEST_F() local
/external/tensorflow/tensorflow/python/framework/
Dtensor_util_test.py1233 def reshape(): function
1249 def reshape(): function

123