| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | canonicalize_all_gather_for_cse_test.cc | 66 const HloInstruction* const reshape = in TEST_F() local
86 const HloInstruction* const reshape = in TEST_F() local
105 const HloInstruction* const reshape = in TEST_F() local
124 const HloInstruction* const reshape = in TEST_F() local
143 const HloInstruction* const reshape = in TEST_F() local
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/generated/spec_V1_0/ |
| D | reshape.example.cpp | 7 namespace generated_tests::reshape { namespace
64 namespace generated_tests::reshape { namespace
155 namespace generated_tests::reshape { namespace
212 namespace generated_tests::reshape { namespace
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | space_to_batch_converter_test.cc | 57 auto reshape = root->operand(0)->operand(0); in TEST_F() local
94 auto reshape = root->operand(0)->operand(0); in TEST_F() local
200 auto reshape = root->operand(0)->operand(0); in TEST_F() local
237 auto reshape = root->operand(0)->operand(0); in TEST_F() local
|
| H A D | dynamic_dimension_simplifier.cc | 93 StatusOr<bool> ReshapeBroadcastForwarding(HloInstruction* reshape) { in ReshapeBroadcastForwarding()
121 StatusOr<bool> ReshapeReshapeForwarding(HloInstruction* reshape) { in ReshapeReshapeForwarding()
153 StatusOr<bool> IdentityReshapeRemoving(HloInstruction* reshape) { in IdentityReshapeRemoving()
|
| H A D | dynamic_padder.cc | 316 HloInstruction* reshape, int64_t input_dim, in GenerateBinaryMask()
469 HloInstruction* reshape, int64_t input_dim, in RewriteDynamicReshapeSplitInput()
652 HloInstruction* reshape, absl::Span<const int64_t> input_dims, in RewriteDynamicReshapeCombineInput()
770 HloInstruction* reshape, absl::Span<const int64_t> input_dims, in RewriteDynamicReshapeSingleGroup()
1479 HloInstruction* reshape = binary->AddInstruction( in RewriteDynamicBinaryOp() local
1647 HloInstruction* reshape, in RewriteDynamicReshape()
|
| H A D | hlo_instruction_test.cc | 1307 HloInstruction* reshape = in TEST_F() local
1406 HloInstruction* reshape = in TEST_F() local
1538 HloInstruction* reshape = in TEST_F() local
1785 HloInstruction* reshape = in TEST_F() local
1832 HloInstruction* reshape = in TEST_F() local
1888 HloInstruction* reshape = in TEST_F() local
|
| H A D | hlo_computation_test.cc | 530 HloInstruction* reshape = in TEST_F() local
566 HloInstruction* reshape = in TEST_F() local
603 HloInstruction* reshape = in TEST_F() local
|
| H A D | dynamic_dimension_inference_test.cc | 581 auto* reshape = builder.AddInstruction( in TEST_F() local
616 auto* reshape = builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F() local
642 auto* reshape = builder.AddInstruction( in TEST_F() local
|
| H A D | reshape_decomposer.cc | 28 Status HandleReshape(HloInstruction* reshape) override { in HandleReshape()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/training/ |
| H A D | saver.py | 342 reshape, argument
389 restore_sequentially, reshape): argument
446 reshape=False, argument
499 reshape=False, argument
781 reshape=False, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | partially_decluster_pass_test.cc | 309 Output reshape = ops::Reshape(s.WithOpName("reshape"), reshape_input, shape); in TEST() local
336 Output reshape = ops::Reshape(s.WithOpName("reshape"), reshape_input, shape); in TEST() local
361 Output reshape = ops::Reshape(s.WithOpName("reshape"), reshape_input, shape); in TEST() local
386 Output reshape = ops::Reshape(s.WithOpName("reshape"), reshape_input, shape); in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Analysis/ |
| H A D | shape_component_analysis.cc | 91 } else if (auto reshape = in visit() local
150 } else if (auto reshape = value.getDefiningOp<mhlo::ReshapeOp>()) { in visit() local
177 } else if (auto reshape = in visit() local
225 } else if (auto reshape = value.getDefiningOp<mhlo::ReshapeOp>()) { in visit() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | const_analysis_test.cc | 157 Output reshape = ops::Reshape(root, arg1, add); in TEST() local
181 Output reshape = ops::Reshape(root, arg1, add); in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | bcast.h | 35 const gtl::InlinedVector<int64_t, 4>& reshape, in ComputeBatchIndices()
94 const Vec& reshape(int i) const { return reshape_[i]; } in reshape() function
|
| H A D | strided_slice_op.h | 133 const Vec& reshape() const { return reshape_; } in reshape() function
|
| /aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| H A D | TensorBase.h | 883 reshape(const NewDimensions& newDimensions) const { in reshape() function
1055 reshape(const NewDimensions& newDimensions) const { in reshape() function
1060 reshape(const NewDimensions& newDimensions) { in reshape() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | scoped_allocator_ops_test.cc | 145 bool reshape, const string& name, int32_t id, in BuildNodeDefWithReshape()
162 void MakeOp(const TensorShape& shape, DataType dtype, bool reshape, in MakeOp()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | literal_test.cc | 667 auto reshape = original.Reshape(/*dimensions=*/{}).value(); in TEST_F() local
686 auto reshape = original.Reshape({3, 4, 2}).value(); in TEST_F() local
706 auto reshape = original.Reshape({3, 4, 2}).value(); in TEST_F() local
713 auto reshape = original.Transpose(/*permutation=*/{}); in TEST_F() local
726 auto reshape = original.Transpose(/*permutation=*/{2, 3, 0, 1}); in TEST_F() local
739 auto reshape = original.Transpose(/*permutation=*/{1, 0}); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | tensor_util_test.py | 1233 def reshape(): function
1249 def reshape(): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | slice_test.cc | 173 auto reshape = Reshape(original, {24, 3, 2, 7}); in TEST_F() local
183 auto reshape = Reshape(original, {2 * 3 * 5, 7}); in TEST_F() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | ReshapeLayerFixture.h | 71 FunctionType reshape; in compute_target() local
|
| /aosp_15_r20/external/armnn/src/armnn/optimizations/ |
| H A D | PermuteAsReshape.hpp | 27 …auto reshape = graph.InsertNewLayer<ReshapeLayer>(permute.GetInputSlot(0), descriptor, name.c_str(… in Run() local
|
| H A D | TransposeAsReshape.hpp | 27 …auto reshape = graph.InsertNewLayer<ReshapeLayer>(transpose.GetInputSlot(0), descriptor, name.c_st… in Run() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | gpu_reduce_scatter_creator.cc | 94 HloInstruction *reshape = nullptr; in Run() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| H A D | get_item_op.cc | 67 auto reshape = xla::Reshape(slice, dims, out_sizes); in Compile() local
|