| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| 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
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| 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
|
| 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()
|
| D | reshape_decomposer.cc | 28 Status HandleReshape(HloInstruction* reshape) override { in HandleReshape()
|
| /external/armnn/src/armnn/optimizations/ |
| D | PermuteAsReshape.hpp | 27 …auto reshape = graph.InsertNewLayer<ReshapeLayer>(permute.GetInputSlot(0), descriptor, name.c_str(… in Run() local
|
| D | TransposeAsReshape.hpp | 27 …auto reshape = graph.InsertNewLayer<ReshapeLayer>(transpose.GetInputSlot(0), descriptor, name.c_st… in Run() local
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | ReshapeLayerFixture.h | 71 FunctionType reshape; in compute_target() local
|
| /external/tensorflow/tensorflow/python/training/ |
| D | saver.py | 342 reshape, argument
389 restore_sequentially, reshape): argument
446 reshape=False, argument
499 reshape=False, argument
781 reshape=False, argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_reduce_scatter_creator.cc | 94 HloInstruction *reshape = nullptr; in Run() local
|
| D | reduction_degenerate_dim_remover.cc | 77 std::unique_ptr<HloInstruction> reshape = in HandleReduce() local
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | get_item_op.cc | 67 auto reshape = xla::Reshape(slice, dims, out_sizes); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| 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
|
| /external/tensorflow/tensorflow/core/util/ |
| D | strided_slice_op.h | 133 const Vec& reshape() const { return reshape_; } in reshape() function
|
| 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
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| 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
|
| /external/armnn/src/backends/backendsCommon/test/ |
| D | ReshapeEndToEndTestImpl.hpp | 32 IConnectableLayer* reshape = network->AddReshapeLayer(descriptor, "reshape"); in CreateReshapeNetwork() local
|
| /external/webrtc/sdk/objc/components/renderer/opengl/ |
| D | RTCNSGLVideoView.m | 80 - (void)reshape { method in RTCNSGLVideoView
|
| /external/eigen/test/ |
| D | reshape.cpp | 196 EIGEN_DECLARE_TEST(reshape) in EIGEN_DECLARE_TEST() argument
|
| /external/clang/test/Analysis/ |
| D | malloc-interprocedural.c | 77 static char *reshape(char *in) { in reshape() function
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | unravel_index_op.cc | 142 Eigen::array<Eigen::Index, 2> reshape{ in Compute() local
|
| 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()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/ |
| D | restrict_max_rank.cc | 160 auto reshape = rewriter.create<ReshapeOp>(op.getLoc(), reshapedTy, input); in matchAndRewrite() local
|
| /external/pytorch/torch/onnx/ |
| D | symbolic_opset14.py | 61 def reshape(g: jit_utils.GraphContext, self, shape): function
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Analysis/ |
| 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
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reshape.cc | 28 namespace reshape { namespace
|