Home
last modified time | relevance | path

Searched refs:output_shapes (Results 1 – 25 of 50) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/
Dbroadcast_to_gpu_kernel.h53 auto output_shapes = AnfAlgo::GetOutputInferShape(kernel_node, 0); in Init() local
54 is_null_input_ = CHECK_NULL_INPUT(input_shapes) || CHECK_NULL_INPUT(output_shapes); in Init()
60 if (input_shapes.size() > SHAPE_SIZE || output_shapes.size() > SHAPE_SIZE) { in Init()
64 if (output_shapes.size() < input_shapes.size()) { in Init()
68 size_t offset = output_shapes.size() - input_shapes.size(); in Init()
73 for (size_t j = 0; j < output_shapes.size(); j++) { in Init()
74 output_shape_[j] = output_shapes[j]; in Init()
Dunsorted_segment_min_gpu_kernel.h54 auto output_shapes = AnfAlgo::GetOutputRealDeviceShapeIfExist(kernel_node, 0); in Init() local
56 …ULL_INPUT(input_shapes) || CHECK_NULL_INPUT(segment_ids_shapes) || CHECK_NULL_INPUT(output_shapes); in Init()
69 if (output_shapes.size() < 1) { in Init()
72 << output_shapes.size() << "."; in Init()
74 num_segments_ = output_shapes[0]; in Init()
86 for (size_t i = 0; i < output_shapes.size(); i++) { in Init()
87 output_size_ *= output_shapes[i]; in Init()
Dunsorted_segment_max_gpu_kernel.h60 auto output_shapes = AnfAlgo::GetOutputRealDeviceShapeIfExist(kernel_node, 0); in Init() local
62 …ULL_INPUT(input_shapes) || CHECK_NULL_INPUT(segment_ids_shapes) || CHECK_NULL_INPUT(output_shapes); in Init()
75 if (output_shapes.size() < 1) { in Init()
78 << output_shapes.size() << "."; in Init()
80 num_segments_ = output_shapes[0]; in Init()
92 for (size_t i = 0; i < output_shapes.size(); i++) { in Init()
93 output_size_ *= output_shapes[i]; in Init()
Dunsorted_segment_sum_gpu_kernel.h58 auto output_shapes = AnfAlgo::GetOutputRealDeviceShapeIfExist(kernel_node, 0); in Init() local
59 … CHECK_NULL_INPUT(input_shapes) || CHECK_NULL_INPUT(ids_shapes) || CHECK_NULL_INPUT(output_shapes); in Init()
82 output_dim0_ = output_shapes[0]; in Init()
83 for (size_t j = 1; j < output_shapes.size(); j++) { in Init()
84 output_dim1_ *= output_shapes[j]; in Init()
Dcast_gpu_kernel.h59 auto output_shapes = AnfAlgo::GetOutputInferShape(kernel_node, 0); in Init() local
60 is_null_input_ = CHECK_NULL_INPUT(input_shapes) || CHECK_NULL_INPUT(output_shapes); in Init()
72 for (size_t j = 0; j < output_shapes.size(); j++) { in Init()
73 output_size_ *= output_shapes[j]; in Init()
Dtopk_gpu_kernel.h88 auto output_shapes = AnfAlgo::GetOutputInferShape(kernel_node, 0); in Init() local
89 is_null_input_ = CHECK_NULL_INPUT(input_shapes) || CHECK_NULL_INPUT(output_shapes); in Init()
100 k_ = output_shapes[output_shapes.size() - 1]; in Init()
/third_party/mindspore/tests/ut/python/dataset/
Dtest_datasets_get_dynamic_shape.py36 dynamic_shapes = dataset.output_shapes()
58 dynamic_shapes = dataset.output_shapes()
75 dynamic_shapes = dataset.output_shapes()
99 dynamic_shapes = dataset.output_shapes()
118 dynamic_shapes = dataset.output_shapes()
142 dynamic_shapes = dataset.output_shapes()
/third_party/mindspore/mindspore/lite/tools/optimizer/fisson/
Dnode_out_shapes.cc31 std::vector<ShapeVector> output_shapes; in Run() local
62 output_shapes.push_back(shape_ptr->cast<abstract::ShapePtr>()->shape()); in Run()
66 output_shapes.push_back(shape); in Run()
70 Spliter::GetInstance()->UpdateNodeOutputShapes(node_name, output_shapes); in Run()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dsingle_kernel_graph.cc24 const std::vector<TypeId> &output_dtypes, const std::vector<std::vector<size_t>> &output_shapes) { in ConstructKernelGraphBasedOnSingleOp() argument
47 if (output_dtypes.size() != output_shapes.size()) { in ConstructKernelGraphBasedOnSingleOp()
50 AnfAlgo::SetOutputInferTypeAndShape(output_dtypes, output_shapes, cnode.get()); in ConstructKernelGraphBasedOnSingleOp()
Dsingle_kernel_graph.h34 … const std::vector<TypeId> &output_dtypes, const std::vector<std::vector<size_t>> &output_shapes);
/third_party/mindspore/tests/
Ddataset_mock.py25 np_types=None, output_shapes=None, input_indexs=()): argument
30 self._output_shapes = output_shapes
46 def output_shapes(self): member in MindData
Dops_common.py260 def build_backward_graph(net, output_shapes, inputs, execute=True): argument
261 inputs = append_sens_to_inputs(output_shapes, inputs)
284 def gen_backward_inputs(input_shapes, output_shapes, config): argument
290 sens_shape = output_shapes[0]
295 def append_sens_to_inputs(output_shapes, inputs): argument
297 sens = Tensor(np.random.normal(0, 1, output_shapes).astype(np.float32))
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dinsert_cast_gpu.cc111 std::vector<std::vector<size_t>> output_shapes; in Run() local
114 (void)output_shapes.emplace_back(shape); in Run()
116 AnfAlgo::SetOutputInferTypeAndShape(output_types, output_shapes, node.get()); in Run()
/third_party/mindspore/tests/ut/python/communication/
Dtest_data_parallel_resnet.py246 np_types=None, output_shapes=None, input_indexs=()): argument
248 np_types=np_types, output_shapes=output_shapes,
284 output_shapes=dataset_shapes,
306 output_shapes=dataset_shapes,
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dlaunch_mul.cc33 std::vector<std::vector<size_t>> output_shapes = {{static_cast<size_t>(shape)}}; in ObtainMulKernelGraph() local
35 kMulOpName, input_dtypes, input_shapes, output_dtypes, output_shapes); in ObtainMulKernelGraph()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/
Dlstm_cpu_kernel.cc46 std::vector<std::vector<size_t>> output_shapes; in InitInputOutputSize() local
49 (void)output_shapes.emplace_back(shape); in InitInputOutputSize()
52 output_shapes[kOutputWorkSpaceIndex] = {len, 1}; in InitInputOutputSize()
53 AnfAlgo::SetOutputInferTypeAndShape(output_types, output_shapes, kernel_node.get()); in InitInputOutputSize()
/third_party/mindspore/tests/ut/python/train/
Dtest_training.py87 output_shapes=dataset_shapes,
141 output_shapes=dataset_shapes,
171 output_shapes=dataset_shapes,
260 output_shapes=dataset_shapes,
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/other/
Dgpu_convert_to_dynamic_shape_gpu_kernel.h60 std::vector<std::vector<size_t>> output_shapes = {input_shape_}; in PostExecute() local
61 AnfAlgo::SetOutputInferTypeAndShape(output_types, output_shapes, kernel_node_.lock().get()); in PostExecute()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_launch_atomic_clean.cc87 std::vector<std::vector<size_t>> output_shapes = {}; in ObtainAtomicCleanKernelGraph() local
89 kAtomicAddrCleanOpName, input_dtypes, input_shapes, output_dtypes, output_shapes); in ObtainAtomicCleanKernelGraph()
Dascend_launch_transdata.cc86 std::vector<std::vector<size_t>> output_shapes = {{shape_}}; in ObtainTransDataKernelGraph() local
88 kTransDataOpName, input_dtypes, input_shapes, output_dtypes, output_shapes); in ObtainTransDataKernelGraph()
/third_party/mindspore/tests/st/pynative/loss_scale/
Dtest_loss_scale.py38 np_types=None, output_shapes=None, input_indexes=(), func_name=''): argument
43 self._output_shapes = output_shapes
60 def output_shapes(self): member in MindData
104 output_shapes=dataset_shapes,
/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Dspliter.cc143 …dateNodeOutputShapes(const std::string &node_name, const std::vector<ShapeVector> &output_shapes) { in UpdateNodeOutputShapes() argument
144 graph_node_output_shapes_[node_name] = (output_shapes); in UpdateNodeOutputShapes()
/third_party/mindspore/tests/ut/python/pipeline/infer/
Dinfer.py57 output_shapes=dataset_shapes,
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/text/kernels/
Dto_number_op.h54 …pe(const std::vector<TensorShape> &input_shapes, std::vector<TensorShape> &output_shapes) override;
Dto_number_op.cc81 …utputShape(const std::vector<TensorShape> &input_shapes, std::vector<TensorShape> &output_shapes) { in OutputShape() argument
82 (void)std::copy(input_shapes.begin(), input_shapes.end(), std::back_inserter(output_shapes)); in OutputShape()

12