Home
last modified time | relevance | path

Searched defs:in_batch (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dreverse_sequence_fp32.c27 const float *in_batch = in + batch * para->input_stride_[para->batch_axis_]; in ReverseSequence() local
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/common/
Dpack_tests.cc65 int in_batch = conv_param->input_batch_; in TEST_F() local
128 int in_batch = conv_param->input_batch_; in TEST_F() local
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Dconv2d_3x3_int8_coder.cc82 int in_batch = conv_param_->input_batch_; in InitTmpBuffer() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dspace_to_batch_int8.c67 int in_batch = i % in_b; in DoSpaceToBatchPaddingNHWCInt8() local
Dpack_int8.c813 int in_batch = conv_param->input_batch_; in PackInputToC8Int8() local