Home
last modified time | relevance | path

Searched defs:output_dtype (Results 1 – 25 of 45) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/tests/ut/python/dataset_deprecated/
Dtest_to_tensor.py145 def test_config(my_output_type, output_dtype): argument
179 def test_config(my_output_type, output_dtype): argument
213 def test_config(my_output_type, output_dtype): argument
247 def test_config(my_output_type, output_dtype): argument
281 def test_config(my_output_type, output_dtype): argument
315 def test_config(my_output_type, output_dtype): argument
/third_party/mindspore/mindspore-src/source/tests/ut/python/dataset/
Dtest_to_tensor.py147 def test_config(my_output_type, output_dtype, check_image=True): argument
182 def test_config(my_output_type, output_dtype, result_output_type=None): argument
232 def test_config(my_output_type, output_dtype): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/add_attr_to_node/
Dimag_fusion.cc35 auto output_dtype = node->Type(); in ImagFusionProcess() local
Dtensorshape_for_ge.cc39 TypeId output_dtype = common::AnfAlgo::GetOutputInferDataType(cnode, 0); in TensorShapeAddDtype() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/opencl/kernel/
Dcast.cc57 auto output_dtype = out_tensors_.front()->data_type(); in CheckSpecs() local
Dcrop.cc46 auto output_dtype = out_tensors_.front()->data_type(); in CheckSpecsWithoutShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dadaptive_max_pool_3d_grad_cpu_kernel.h57 TypeId output_dtype; variable
Dembedding_look_up_cpu_kernel.cc30 #define ADD_KERNEL(input_params_dtype, input_indices_dtype, output_dtype, input_params_type, input_… argument
40 #define ADD_KERNEL_INT32(input_params_dtype, input_indices_dtype, output_dtype, input_params_type, … argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dstandard_laplace.cc32 auto output_dtype = ctx.Output(0)->GetDataType(); in Compute() local
Drandom_uniform_int.cc34 auto output_dtype = output->GetDataType(); in Compute() local
Dtrace.cc66 auto output_dtype = output_tensor->GetDataType(); in Compute() local
Dbartlett_window.cc37 auto output_dtype = ctx.Output(0)->GetDataType(); in Compute() local
Dcauchy.cc47 auto output_dtype = output_tensor->GetDataType(); in Compute() local
Dgamma.cc33 auto output_dtype = output->GetDataType(); in Compute() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/
Dtest_ops_diag.cc38 TypePtr output_dtype; member
Dtest_ops_sinh.cc53 auto output_dtype = sinh_func_impl.InferType(prim, {input}); in TEST_P() local
Dtest_ops_argmax_ext.cc33 TypePtr output_dtype; member
Dtest_ops_pad_constant.cc32 TypePtr output_dtype; member
Dtest_ops_pad_Reflect_Replicate.cc36 TypePtr output_dtype; member
Dtest_ops_eig.cc40 TypePtr output_dtype; member
Dtest_ops_ones.cc32 TypePtr output_dtype; member
Dtest_ops_zeros.cc32 TypePtr output_dtype; member
Dtest_ops_fft_with_size.cc34 TypePtr output_dtype; member
Dtest_ops_dct.cc35 TypePtr output_dtype; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/random/
Drandom_poisson_gpu_kernel.cc32 #define ADD_KERNEL(shape_dtype, rate_dtype, output_dtype, rate_type, output_type) \ argument

12