Home
last modified time | relevance | path

Searched refs:input_arg (Results 1 – 14 of 14) sorted by relevance

/third_party/mindspore/mindspore/core/ops/
Ddynamic_broadcast_gradient_args.cc30 int64_t CheckInputsAndGetShape(const AbstractBasePtr &input_arg, const string &prim_name) { in CheckInputsAndGetShape() argument
31 MS_EXCEPTION_IF_NULL(input_arg); in CheckInputsAndGetShape()
32 if (input_arg->isa<abstract::AbstractTensor>()) { in CheckInputsAndGetShape()
33 …auto input_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_arg->BuildShape())[kShape… in CheckInputsAndGetShape()
39 …auto max_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_arg->BuildShape())[kMaxShap… in CheckInputsAndGetShape()
46 } else if (input_arg->isa<abstract::AbstractTuple>()) { in CheckInputsAndGetShape()
47 auto x_shape = dyn_cast<abstract::AbstractTuple>(input_arg); in CheckInputsAndGetShape()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Drelux_int8_coder.cc29 quant_arg_.input_arg.scale_ = input_tensor_->quant_params().front().scale; in Prepare()
30 quant_arg_.input_arg.zp_ = input_tensor_->quant_params().front().zeroPoint; in Prepare()
34 const double multiplier = quant_arg_.input_arg.scale_ / quant_arg_.output_arg.scale_; in Prepare()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Drelux_int8.cc35 quant_arg_.input_arg.scale_ = input->quant_params().front().scale; in Init()
36 quant_arg_.input_arg.zp_ = input->quant_params().front().zeroPoint; in Init()
40 const double multiplier = quant_arg_.input_arg.scale_ / quant_arg_.output_arg.scale_; in Init()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Drelux_int8.c20 if (src[i] <= arg->input_arg.zp_) { in ReluXInt8()
24 const int32_t input_val = src[i] - arg->input_arg.zp_; in ReluXInt8()
Drelux_int8.h26 QuantArg input_arg; member
/third_party/boost/boost/compute/algorithm/detail/
Dinplace_reduce.hpp53 size_t input_arg = k.add_arg<value_type *>(memory_object::global_memory, "input"); in inplace_reduce() local
101 kernel.set_arg(input_arg, *input_buffer); in inplace_reduce()
/third_party/boost/libs/compute/include/boost/compute/algorithm/detail/
Dinplace_reduce.hpp53 size_t input_arg = k.add_arg<value_type *>(memory_object::global_memory, "input"); in inplace_reduce() local
101 kernel.set_arg(input_arg, *input_buffer); in inplace_reduce()
/third_party/node/test/pseudo-tty/
Dtestcfg.py43 def __init__(self, path, file, expected, input_arg, arch, mode, context, config): argument
47 self.input = input_arg
/third_party/mindspore/mindspore/core/utils/
Dcheck_convert_utils.cc387 auto input_arg = input_args[index]; in GetInputTensorType() local
388 if (input_arg == nullptr) { in GetInputTensorType()
391 auto base_type = input_arg->BuildType(); in GetInputTensorType()
/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dtf_model_parser.cc611 auto &input_arg = tf_sub_signature.input_arg(j); in ConvertSubgraphInputs() local
614 parameter->set_name(input_arg.name()); in ConvertSubgraphInputs()
615 (*anf_sub_node_map)[input_arg.name()] = parameter; in ConvertSubgraphInputs()
/third_party/mbedtls/tests/suites/
Dtest_suite_psa_crypto_driver_wrappers.function777 data_t *input_arg,
805 input_buffer_size = ( (size_t) input_arg->len + (size_t) iv->len );
810 memcpy( input + iv->len, input_arg->x, input_arg->len );
Dtest_suite_psa_crypto.function3284 data_t *input_arg,
3312 input_buffer_size = ( (size_t) input_arg->len + (size_t) iv->len );
3317 memcpy( input + iv->len, input_arg->x, input_arg->len );
3340 data_t *input_arg,
3360 input_buffer_size = ( (size_t) input_arg->len + (size_t) iv->len );
3365 memcpy( input + iv->len, input_arg->x, input_arg->len );
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/serializers/nnacl_serializer/
Dnnacl_int8_serializer.cc230 CodeBaseStruct("ReluXQuantArg", name, relu_quant_arg.input_arg, relu_quant_arg.output_arg, in CodeStruct()
/third_party/mindspore/third_party/proto/tensorflow/
Dop_def.proto52 repeated ArgDef input_arg = 2; field