| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | tensor_list_reserve.cc | 28 void TensorListReserve::Init(const int64_t element_dtype, const int64_t shape_type) { in Init()
37 void TensorListReserve::set_shape_type(const int64_t shape_type) { in set_shape_type()
|
| D | tensor_list_from_tensor.cc | 27 void TensorListFromTensor::Init(const int64_t element_dtype, const int64_t shape_type) { in Init()
46 void TensorListFromTensor::set_shape_type(const int64_t shape_type) { in set_shape_type()
|
| D | expand.cc | 116 auto shape_type = input_args[1]->GetType(); in ExpandInferShape() local
152 TypePtr shape_type = input_args[1]->GetType(); in ExpandInferType() local
|
| D | sparse_reorder.cc | 72 auto shape_type = input_args[2]->GetType(); in SparseReorderInferType() local
|
| D | non_deterministic_ints.cc | 77 auto shape_type = input_args[0]->GetType(); in NonDeterministicIntsInferType() local
|
| D | sparse_tensor_dense_add.cc | 101 auto shape_type = input_args[kInputIndex2]->GetType(); in SparseTensorDenseAddInferType() local
|
| D | sparse_addmm.cc | 94 TypePtr shape_type = input_args[kInputIndex2]->GetType(); in SparseAddmmInferType() local
|
| D | sparse_slice.cc | 95 auto shape_type = input_args[kInputIndex2]->GetType(); in InferType() local
|
| D | ragged_tensor_to_tensor.cc | 113 TypePtr shape_type = input_args[kInputIndex0]->GetType(); in RaggedTensorToTensorInferType() local
|
| D | sparse_split.cc | 110 auto shape_type = input_args[kInputIndex3]->GetType(); in SparseSplitInferType() local
|
| D | sparse_tensor_dense_mat_mul.cc | 228 TypePtr shape_type = input_args[2]->GetType(); in SparseTensorDenseMatmulInferType() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | dropout_gen_mask_ext.cc | 55 auto shape_type = input_args[kIndex0]->GetType(); in CheckValidation() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | trace_grad.cc | 50 DataType shape_type = ctx.Input(1)->GetDataType(); in Compute() local
|
| D | scatter_nd.cc | 124 auto shape_type = static_cast<DataType>(ctx.Input(2)->GetDataType()); in DTYPE_CHOOSE() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/ |
| D | aicpu_kernel_build.cc | 51 auto *shape_type = reinterpret_cast<int32_t *>(ext_info_buf + ext_info_offset); in SetExtInfoShapeType() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | max_pool_grad_with_argmax_v2_cpu_kernel.cc | 155 #define ADD_KERNEL(x_dtype, shape_dtype, x_type, shape_type) \ argument
|
| D | max_pool_with_argmax_v2_cpu_kernel.cc | 161 #define ADD_KERNEL(x_dtype, shape_dtype, x_type, shape_type) \ argument
|
| D | max_pool3d_with_argmax_cpu_kernel.cc | 277 #define ADD_KERNEL(x_dtype, shape_dtype, x_type, shape_type) \ argument
|
| D | max_pool3d_grad_with_argmax_cpu_kernel.cc | 254 #define ADD_KERNEL(x_dtype, shape_dtype, x_type, shape_type) \ argument
|