| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/ |
| D | trt_op_converter.cc | 145 …*(inputs[0].tensor()), pooling_type, nvinfer1::DimsHW{LongToInt(kernel_size[2]), LongToInt(kernel_… in AddPoolingLayer()
149 layer->setStride(nvinfer1::DimsHW{LongToInt(strides[2]), LongToInt(strides[3])}); in AddPoolingLayer()
206 int offset = axis[i] >= 0 ? LongToInt(axis[i]) : LongToInt(axis[i] + input_shape.size()); in AddReduceLayer()
211 int offset = axis >= 0 ? LongToInt(axis) : LongToInt(axis + input_shape.size()); in AddReduceLayer()
274 *(inputs[0].tensor()), LongToInt(out_channel), in MS_TRT_CONVERTER_FUNC_REG()
275 …nvinfer1::DimsHW{LongToInt(kernel_size[0]), LongToInt(kernel_size[1])}, *(inputs[1].weight()), bia… in MS_TRT_CONVERTER_FUNC_REG()
279 layer->setStride(nvinfer1::DimsHW{LongToInt(strides[2]), LongToInt(strides[3])}); in MS_TRT_CONVERTER_FUNC_REG()
289 layer->setPrePadding(nvinfer1::DimsHW{LongToInt(pad_list[0]), LongToInt(pad_list[2])}); in MS_TRT_CONVERTER_FUNC_REG()
290 layer->setPostPadding(nvinfer1::DimsHW{LongToInt(pad_list[1]), LongToInt(pad_list[3])}); in MS_TRT_CONVERTER_FUNC_REG()
728 …nvinfer1::DimsHW{LongToInt(kernel_size[0]), LongToInt(kernel_size[1])}, *(inputs[1].weight()), bia… in MS_TRT_CONVERTER_FUNC_REG()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/rl/ |
| D | buffer_append_cpu_kernel.h | 72 …int remain_size = (exp_batch_ > (capacity_ - index)) ? LongToInt(capacity_ - index) : LongToInt(ex… in Launch()
73 …int remap_size = (exp_batch_ > (capacity_ - index)) ? LongToInt(exp_batch_ - capacity_ + index) : … in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/rl/ |
| D | buffer_append_gpu_kernel.cc | 75 IncreaseCount(capacity_, LongToInt(exp_batch_), count_addr, head_addr, index_addr, cuda_stream); in Launch()
80 …BufferAppend(capacity_, one_exp_len, index_addr, LongToInt(exp_batch_), buffer_addr, exp_addr, cud… in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | slice_grad_cpu_kernel.cc | 42 [](const int64_t &value) { return LongToInt(value); }); in InitKernel()
50 [](const int64_t &value) { return LongToInt(value); }); in InitKernel()
52 [](const int64_t &value) { return LongToInt(value); }); in InitKernel()
60 [](const int64_t &value) { return LongToInt(value); }); in InitKernel()
|
| D | stridedslice_cpu_kernel.cc | 132 int begin_pos = LongToInt(begin[i]); in InitSliceParam()
133 int end_pos = LongToInt(end[i]); in InitSliceParam()
134 int stride_size = LongToInt(stride[i]); in InitSliceParam()
|
| D | slice_cpu_kernel.cc | 71 int begin_pos = LongToInt(begin[i]); in InitSliceParam()
72 int slice_size = LongToInt(size[i]); in InitSliceParam()
|
| D | random_cpu_kernel.cc | 122 seed_ = LongToInt(GetValue<int64_t>(prim->GetAttr("seed"))); in InitKernel()
123 seed2_ = LongToInt(GetValue<int64_t>(prim->GetAttr("seed2"))); in InitKernel()
|
| D | unpack_cpu_kernel.cc | 39 unstack_param_.axis_ = LongToInt(axis_tmp); in InitKernel()
|
| D | concat_cpu_kernel.cc | 31 axis_ = LongToInt(AnfAlgo::GetNodeAttr<int64_t>(kernel_node, AXIS)); in InitKernel()
|
| D | split_cpu_kernel.cc | 67 param.split_dim_ = LongToInt(axis_); in LaunchSplit()
|
| D | tile_cpu_kernel.cc | 61 [](const int64_t &value) { return LongToInt(value); }); in TileTensorParamrInit()
|
| D | l2_normalize_cpu_kernel.cc | 34 axis_ = LongToInt(AnfAlgo::GetNodeAttr<int64_t>(kernel_node, AXIS)); in InitKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
| D | lstm_cpu_kernel.cc | 121 input_size_ = LongToInt(AnfAlgo::GetNodeAttr<int64_t>(kernel_node, "input_size")); in CheckParam()
122 hidden_size_ = LongToInt(AnfAlgo::GetNodeAttr<int64_t>(kernel_node, "hidden_size")); in CheckParam()
123 num_layers_ = LongToInt(AnfAlgo::GetNodeAttr<int64_t>(kernel_node, "num_layers")); in CheckParam()
|
| D | pooling_avg_grad_cpu_kernel.cc | 47 [](const int64_t &value) { return LongToInt(value); }); in InitKernel()
49 [](const int64_t &value) { return LongToInt(value); }); in InitKernel()
|
| D | conv2d_grad_filter_cpu_kernel.cc | 59 [](const int64_t &value) { return LongToInt(value); }); in InitKernel()
61 [](const int64_t &value) { return LongToInt(value); }); in InitKernel()
|
| D | conv_cpu_kernel.cc | 67 [](const int64_t &value) { return LongToInt(value); }); in InitKernel()
69 [](const int64_t &value) { return LongToInt(value); }); in InitKernel()
|
| D | pooling_max_grad_cpu_kernel.cc | 45 [](const int64_t &value) { return LongToInt(value); }); in InitKernel()
47 [](const int64_t &value) { return LongToInt(value); }); in InitKernel()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | convert_tensor_utils.h | 32 void LongToInt(void *dst, const void *src, size_t elem_num);
|
| D | convert_tensor_utils.cc | 87 void LongToInt(void *dst, const void *src, size_t elem_num) { in LongToInt() function
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/strategy_checkpoint/ |
| D | parallel_strategy_checkpoint.cc | 177 tensor_map->add_dim(LongToInt(map_dim)); in Save()
187 parallel_layouts->set_field(LongToInt(tensor_layout->get_field_size())); in Save()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | softmax_gpu_kernel.h | 127 auto axis = LongToInt(GetAttr<int64_t>(kernel_node, "axis")); in Init()
134 [](const int64_t &value) { return LongToInt(value); }); in Init()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | cpu_device_address.cc | 137 LongToInt(ptr_, host_ptr, size / sizeof(int64_t)); in SyncHostToDevice()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/fl/ |
| D | start_fl_job_kernel.h | 83 data_size_ = LongToInt(AnfAlgo::GetNodeAttr<int64_t>(kernel_node, "data_size")); in Init()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/ |
| D | embedding_look_up_ps_kernel.cc | 95 int index = SizeToInt(lookup_ids[i]) - LongToInt(offset_); in UpdateEmbeddings()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/trt/ |
| D | trt_utils.h | 79 trt_dims.d[i - offset] = LongToInt(ms_shape[i]);
|