Home
last modified time | relevance | path

Searched defs:Reshape (Results 1 – 17 of 17) sorted by relevance

/third_party/mindspore/mindspore/core/ops/
Dreshape.h35 Reshape() : PrimitiveC(kNameReshape) { InitIOName({"tensor", "shape"}, {"output"}); } in Reshape() function
/third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pre_activate/
Dtranspose_reshape_fusion_test.py19 Reshape = P.Reshape() variable
Dreshape_transpose_fusion_test.py19 Reshape = P.Reshape() variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/
Dcumsum_gpu_kernel.h96 void Reshape() { in Reshape() function
Dcumprod_gpu_kernel.h96 void Reshape() { in Reshape() function
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/
Dgather_gpu_kernel.h99 void Reshape() { in Reshape() function
Dgather_grad_gpu_kernel.h100 void Reshape() { in Reshape() function
Dgatherv2_gpu_kernel.h127 void Reshape() { in Reshape() function
Dembedding_lookup_gpu_kernel.h119 void Reshape() { in Reshape() function
Dgathernd_gpu_kernel.h137 void Reshape() { in Reshape() function
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/core/
Dcv_tensor.cc117 Status CVTensor::Reshape(const TensorShape &shape) { in Reshape() function in mindspore::dataset::CVTensor
Dtensor.cc483 Status Tensor::Reshape(const TensorShape &shape) { in Reshape() function in mindspore::dataset::Tensor
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dcumsum_cpu_kernel.cc98 void CumSumCPUKernel::Reshape() { in Reshape() function in mindspore::kernel::CumSumCPUKernel
/third_party/mindspore/mindspore/_extends/graph_kernel/model/
Dop_infer.py218 class Reshape(_Reshape): class
/third_party/mindspore/tests/ut/cpp/dataset/
Dtensor_test.cc87 TEST_F(MindDataTestTensorDE, Reshape) { in TEST_F() argument
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/
Dtrt_op_converter.cc619 MS_TRT_CONVERTER_FUNC_REG(Reshape) { return AddReshapeLayer(node, context); } in MS_TRT_CONVERTER_FUNC_REG() argument
/third_party/mindspore/mindspore/ops/operations/
Darray_ops.py448 class Reshape(PrimitiveWithInfer): class