| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/ |
| D | stack_tests.cc | 36 std::vector<int> input_shapes[INPUT_NUM] = {{10}, {10}}; in TEST_F() local
53 std::vector<int> input_shapes[INPUT_NUM] = {{3, 4}, {3, 4}}; in TEST_F() local
72 std::vector<int> input_shapes[INPUT_NUM] = {{3, 4, 5}, {3, 4, 5}}; in TEST_F() local
91 std::vector<int> input_shapes[INPUT_NUM] = {{3, 4, 5}, {3, 4, 5}}; in TEST_F() local
110 std::vector<int> input_shapes[INPUT_NUM] = {{1, 96}, {1, 96}}; in TEST_F() local
129 std::vector<int> input_shapes[INPUT_NUM] = {{3, 4, 6}, {3, 4, 6}}; in TEST_F() local
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/ |
| D | batchnorm_fp32_coder.cc | 30 std::vector<int> input_shapes = input_tensor_->shape(); in Init() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | launch_mul.cc | 32 std::vector<std::vector<int64_t>> input_shapes = {{shape}, {1}}; in ObtainMulKernelGraph() local
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fisson/ |
| D | node_out_shapes.cc | 30 std::vector<ShapeVector> input_shapes; in Run() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | single_kernel_graph.cc | 23 …ng &op_name, const std::vector<TypeId> &input_dtypes, const std::vector<ShapeVector> &input_shapes, in ConstructKernelGraphBasedOnSingleOp()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | arithmetic_info.cc | 40 Shapes input_shapes; in InferExpendShape() local
80 Shapes input_shapes = InferExpendShape(); in CheckStrategy() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | cast_gpu_kernel.h | 58 auto input_shapes = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 0); in Init() local
|
| D | broadcast_to_gpu_kernel.h | 52 auto input_shapes = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 0); in Init() local
|
| D | unsorted_segment_sum_gpu_kernel.h | 56 auto input_shapes = AnfAlgo::GetInputRealDeviceShapeIfExist(kernel_node, 0); in Init() local
|
| D | topk_gpu_kernel.h | 87 auto input_shapes = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 0); in Init() local
|
| D | unsorted_segment_max_gpu_kernel.h | 58 auto input_shapes = AnfAlgo::GetInputRealDeviceShapeIfExist(kernel_node, 0); in Init() local
|
| D | unsorted_segment_min_gpu_kernel.h | 52 auto input_shapes = AnfAlgo::GetInputRealDeviceShapeIfExist(kernel_node, 0); in Init() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | batchnorm_int8.cc | 172 auto input_shapes = in_tensors_.at(0)->shape(); in Init() local
205 auto input_shapes = in_tensors_.at(0)->shape(); in ReSize() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_launch_transdata.cc | 85 std::vector<std::vector<int64_t>> input_shapes = {{input_shape}}; in ObtainTransDataKernelGraph() local
|
| D | ascend_launch_atomic_clean.cc | 86 std::vector<std::vector<int64_t>> input_shapes = {{static_cast<int64_t>(shape)}}; in ObtainAtomicCleanKernelGraph() local
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/parallel/ |
| D | spliter.cc | 139 …::UpdateNodeInputShapes(const std::string &node_name, const std::vector<ShapeVector> &input_shapes… in UpdateNodeInputShapes()
|
| D | conv2d_info.cc | 154 auto input_shapes = input_shape_iter->second; in CreateOutputsOfSplit() local
|
| D | multi_conv_info.cc | 192 auto input_shapes = input_shape_iter->second; in SplitSingleConv() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | batchnorm_fp32.cc | 64 auto input_shapes = in_tensors_.at(0)->shape(); in FillParam() local
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/text/kernels/ |
| D | to_number_op.cc | 81 Status ToNumberOp::OutputShape(const std::vector<TensorShape> &input_shapes, std::vector<TensorShap… in OutputShape()
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/ |
| D | batchnorm_int8_coder.cc | 28 std::vector<int> input_shapes = input_tensor_->shape(); in Prepare() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/ |
| D | dynamic_broadcast_gradient_args_kernel.cc | 198 std::vector<std::vector<int64_t>> input_shapes(kInputNum); in Execute() local
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | parameter_server.cc | 227 const std::vector<size_t> &input_shapes = lookup->input_sizes(); in InitEmbeddingTable() local
305 … [](const std::shared_ptr<std::vector<size_t>> &input_shapes) -> std::vector<size_t> { in UpdateWeights()
552 const std::vector<size_t> &input_shapes = lookup->input_sizes(); in SyncEmbeddingTables() local
|
| /third_party/ffmpeg/libavfilter/dnn/ |
| D | dnn_backend_openvino.c | 279 input_shapes_t input_shapes; in init_model_ov() local
494 input_shapes_t input_shapes; in get_output_ov() local
|
| /third_party/mindspore/tests/ |
| D | ops_common.py | 277 def gen_inputs(input_shapes, config): argument
284 def gen_backward_inputs(input_shapes, output_shapes, config): argument
|