| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/ir/vision/ |
| D | to_tensor_ir.cc | 24 ToTensorOperation::ToTensorOperation(const std::string &output_type) { in ToTensorOperation()
29 ToTensorOperation::ToTensorOperation(const DataType &output_type) { output_type_ = output_type; } in ToTensorOperation()
55 std::string output_type = op_params["output_type"]; in from_json() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/image/ |
| D | to_tensor_op.h | 30 explicit ToTensorOp(const DataType::Type &output_type) : output_type_(output_type) {} in ToTensorOp()
32 explicit ToTensorOp(const std::string &output_type) : output_type_(DataType(output_type)) {} in ToTensorOp()
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/ |
| D | test_crop_and_resize_grad_image_op.py | 37 def test_crop_and_resize_grad_image_bilinear(input_type, output_type): argument
76 def test_crop_and_resize_grad_image_nearest(input_type, output_type): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/ |
| D | add_akg_kernel_attrs.cc | 47 TypeId output_type = AnfAlgo::GetOutputDeviceDataType(node, 0); in ProcessCast() local
53 TypeId output_type = AnfAlgo::GetOutputDeviceDataType(node, 0); in ProcessMatMul() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/symbol_ops_impl/ |
| D | scalar_cast.cc | 26 auto output_type = b->out_abstract()->GetType()->generic_type_id(); in __anon299a202d0102() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | rand_like_ext.cc | 28 TypePtr output_type; in InferType() local
|
| D | zeros_like_ext.cc | 37 auto output_type = TypeIdToType(static_cast<TypeId>(val)); in InferType() local
|
| D | ones_like_ext.cc | 37 auto output_type = TypeIdToType(static_cast<TypeId>(val)); in InferType() local
|
| D | fill.cc | 61 auto output_type = TypeIdToType(static_cast<TypeId>(val)); in InferType() local
|
| D | cumsum_ext.cc | 68 auto output_type = TypeIdToType(static_cast<TypeId>(val)); in InferType() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/dyn_shape_dev/ |
| D | test_argmax.py | 22 def argmax_(input_x, axis, output_type): argument
|
| D | test_argmin.py | 22 def argmin_(input_x, axis, output_type): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/environ/ |
| D | environ_cpu_destroy_all.cc | 28 auto output_type = outputs[kIndex0]->dtype_id(); in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/environ/ |
| D | environ_gpu_destroy_all.cc | 25 auto output_type = outputs[kIndex0]->dtype_id(); in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/ |
| D | py_func_op.h | 34 explicit PyFuncOp(py::function func, DataType::Type output_type) in PyFuncOp()
|
| /third_party/grpc/src/compiler/ |
| D | php_generator.cc | 70 const Descriptor* output_type = method->output_type(); in PrintMethod() local
141 const Descriptor* output_type = method->output_type(); in PrintServerMethod() local
|
| D | node_generator.cc | 103 const Descriptor* output_type = method->output_type(); in GetAllMessages() local
163 const Descriptor* output_type = method->output_type(); in PrintMethod() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | dynamic_broadcast_gradient_args.cc | 92 auto output_type = std::make_shared<Tuple>(types); in InferType() local
102 auto output_type = std::make_shared<Tuple>(types); in InferShapeAndType() local
|
| D | dropout_nd.cc | 128 TypePtr output_type = DropoutNDInferType(primitive, input_args); in Dropout2DInfer() local
135 TypePtr output_type = DropoutNDInferType(primitive, input_args); in Dropout3DInfer() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/format_type/ |
| D | utils.cc | 24 … const std::string &format, const TypeId &input_type, const TypeId &output_type, in AddCastOpNodeToGraph()
|
| D | remove_host_kernel.cc | 39 auto output_type = TypeId::kNumberTypeInt64; in Process() local
|
| /third_party/mesa3d/src/freedreno/ir3/ |
| D | ir3_shared_folding.c | 33 type_t output_type = ir3_output_conv_type(src, &can_fold); in try_shared_folding() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/ |
| D | tensor_copy_gpu_kernel.cc | 37 auto output_type = outputs.at(kIndex0)->dtype_id(); in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/optimizer/ |
| D | cpu_pass_utils.cc | 38 const TypeId &input_type, const TypeId &output_type, in AddCastOpNodeToGraph()
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/ |
| D | test_ops_scalar_arithmetic.cc | 49 TypePtr output_type; member
|