| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | sparse_sparse_minimum_impl.cu | 93 S *output_values, int64_t limit1) { in SparseSparseMinimum3()
122 half *output_values, int64_t limit1) { in SparseSparseMinimum3()
148 S *output_values, const int64_t rank_1, int64_t *sum_ptr) { in Min_test1()
160 half *output_values, const int64_t rank_1, int64_t *sum_ptr) { in Min_test1()
|
| D | sparse_sparse_maximum_impl.cu | 91 S *output_values, int64_t limit1) { in SparseSparseMaximum3()
119 half *output_values, int64_t limit1) { in SparseSparseMaximum3()
145 S *output_values, const int64_t rank_1, int64_t *sum_ptr) { in Max_test1()
157 half *output_values, const int64_t rank_1, int64_t *sum_ptr) { in Max_test1()
|
| D | sparse_fill_empty_rows.cu | 36 … Complex<S> *output_values, int64_t *output_indices, int indice_num, in AssignValueKernel()
167 … S *default_value, int *emptyrow_count, int64_t *input_row_end, S *output_values, in AssignValueKernel()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | sparse_slice_cpu_kernel.cc | 136 auto output_values = static_cast<T *>(outputs[kIndex1]->device_ptr()); in LaunchKernel() local
147 … int64_t *start, int64_t *size, int64_t *output_indices, T *output_values, in SliceCompute()
|
| D | sparse_concat_cpu_kernel.cc | 126 auto output_values = static_cast<S *>(outputs[kOutputValuesStart]->device_ptr()); in SparseConcat() local
|
| D | sparse_count_sparse_output_cpu_kernel.cc | 156 auto *output_values = reinterpret_cast<T *>(outputs[1]->device_ptr()); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/ |
| D | median_proto.cc | 24 … const string &attr3_name, const string &output_values, const string &output_indices) { in InferShapeAndTypeMedian()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | sparse_slice.cc | 37 void Slice(Tensor *output_indices, Tensor *output_values, Tensor *output_dense_shape, SparseTensor … in Slice()
195 Tensor *output_values = ctx.Output(1); in Compute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/jit/ |
| D | jit_dfunctor.cc | 100 std::vector<ValuePtr> output_values; in GenNewTensor() local
115 std::vector<ValuePtr> output_values; in GenNewTensor() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/ |
| D | sparse_sparse_maximum_cpu_kernel.cc | 284 …::TensorMap<Eigen::Tensor<T, 1, Eigen::RowMajor, Eigen::DenseIndex>, Eigen::Aligned> output_values( in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/forward/ |
| D | forward.cc | 394 std::vector<ValuePtr> output_values; in CreateViewOpOutputs() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/ |
| D | session_basic.cc | 662 ValuePtrList output_values; in HandleOpOutputs() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/ |
| D | pynative_utils.cc | 1664 std::vector<ValuePtr> output_values(size); in MakeOutputValue() local
|