| /third_party/mindspore/tests/ |
| D | dataset_mock.py | 25 np_types=None, output_shapes=None, input_indexs=()): argument
46 def output_shapes(self): member in MindData
|
| D | ops_common.py | 260 def build_backward_graph(net, output_shapes, inputs, execute=True): argument
284 def gen_backward_inputs(input_shapes, output_shapes, config): argument
295 def append_sens_to_inputs(output_shapes, inputs): argument
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | launch_mul.cc | 33 std::vector<std::vector<size_t>> output_shapes = {{static_cast<size_t>(shape)}}; in ObtainMulKernelGraph() local
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fisson/ |
| D | node_out_shapes.cc | 31 std::vector<ShapeVector> output_shapes; in Run() local
|
| D | fisson_util.cc | 230 auto output_shapes = Spliter::GetInstance()->graph_node_output_shapes()[out_node_name]; in UpdateSplitInfo() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | single_kernel_graph.cc | 24 const std::vector<TypeId> &output_dtypes, const std::vector<std::vector<size_t>> &output_shapes) { in ConstructKernelGraphBasedOnSingleOp()
|
| /third_party/mindspore/tests/st/pynative/loss_scale/ |
| D | test_loss_scale.py | 38 np_types=None, output_shapes=None, input_indexes=(), func_name=''): argument
60 def output_shapes(self): member in MindData
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | cast_gpu_kernel.h | 59 auto output_shapes = AnfAlgo::GetOutputInferShape(kernel_node, 0); in Init() local
|
| D | broadcast_to_gpu_kernel.h | 53 auto output_shapes = AnfAlgo::GetOutputInferShape(kernel_node, 0); in Init() local
|
| D | unsorted_segment_sum_gpu_kernel.h | 58 auto output_shapes = AnfAlgo::GetOutputRealDeviceShapeIfExist(kernel_node, 0); in Init() local
|
| D | topk_gpu_kernel.h | 88 auto output_shapes = AnfAlgo::GetOutputInferShape(kernel_node, 0); in Init() local
|
| D | unsorted_segment_max_gpu_kernel.h | 60 auto output_shapes = AnfAlgo::GetOutputRealDeviceShapeIfExist(kernel_node, 0); in Init() local
|
| D | unsorted_segment_min_gpu_kernel.h | 54 auto output_shapes = AnfAlgo::GetOutputRealDeviceShapeIfExist(kernel_node, 0); in Init() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/other/ |
| D | gpu_convert_to_dynamic_shape_gpu_kernel.h | 60 std::vector<std::vector<size_t>> output_shapes = {input_shape_}; in PostExecute() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_launch_transdata.cc | 86 std::vector<std::vector<size_t>> output_shapes = {{shape_}}; in ObtainTransDataKernelGraph() local
|
| D | ascend_launch_atomic_clean.cc | 87 std::vector<std::vector<size_t>> output_shapes = {}; in ObtainAtomicCleanKernelGraph() local
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/parallel/ |
| D | spliter.cc | 143 …:UpdateNodeOutputShapes(const std::string &node_name, const std::vector<ShapeVector> &output_shape… in UpdateNodeOutputShapes()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | insert_cast_gpu.cc | 111 std::vector<std::vector<size_t>> output_shapes; in Run() local
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/text/kernels/ |
| D | to_number_op.cc | 81 …utputShape(const std::vector<TensorShape> &input_shapes, std::vector<TensorShape> &output_shapes) { in OutputShape()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
| D | lstm_cpu_kernel.cc | 46 std::vector<std::vector<size_t>> output_shapes; in InitInputOutputSize() local
|
| /third_party/mindspore/tests/ut/python/communication/ |
| D | test_data_parallel_resnet.py | 246 np_types=None, output_shapes=None, input_indexs=()): argument
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | parameter_server.cc | 393 const std::vector<size_t> output_shapes = table_lookup_op->output_sizes(); in DoEmbeddingLookup() local
|
| /third_party/mindspore/mindspore/dataset/engine/ |
| D | datasets.py | 1547 def output_shapes(self): member in Dataset
3192 def output_shapes(self): member in TransferDataset
|