Home
last modified time | relevance | path

Searched refs:prev_output (Results 1 – 12 of 12) sorted by relevance

/external/ComputeLibrary/src/core/CL/kernels/
DCLArgMinMaxLayerKernel.cpp43 Status validate_arguments(const ITensorInfo *input, const ITensorInfo *prev_output, const ITensorIn… in validate_arguments() argument
56 if(prev_output != nullptr && prev_output->total_size() != 0) in validate_arguments()
58 …ARM_COMPUTE_RETURN_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(prev_output, 1, DataType::U32, DataType::S32); in validate_arguments()
61 ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_DATA_TYPES(prev_output, output); in validate_arguments()
75 void CLArgMinMaxLayerKernel::configure(const ICLTensor *input, const ICLTensor *prev_output, ICLTen… in configure() argument
77 configure(CLKernelLibrary::get().get_compile_context(), input, prev_output, output, axis, op); in configure()
80 …pileContext &compile_context, const ICLTensor *input, const ICLTensor *prev_output, ICLTensor *out… in configure() argument
88 …COMPUTE_ERROR_THROW_ON(validate_arguments(input->info(), (prev_output != nullptr) ? prev_output->i… in configure()
90 auto padding_info = get_padding_info({ input, prev_output, output }); in configure()
93 _prev_output = prev_output; in configure()
[all …]
DCLArgMinMaxLayerKernel.h67 …void configure(const ICLTensor *input, const ICLTensor *prev_output, ICLTensor *output, unsigned i…
79 …pileContext &compile_context, const ICLTensor *input, const ICLTensor *prev_output, ICLTensor *out…
93 …static Status validate(const ITensorInfo *input, const ITensorInfo *prev_output, const ITensorInfo…
/external/XNNPACK/eval/
Du32-sqrt.cc36 const uint32_t prev_output = output - 1; in TEST() local
37 const int64_t squared_prev_output = int64_t(uint64_t(prev_output) * uint64_t(prev_output)); in TEST()
80 const uint32_t prev_output = output - 1; in TEST() local
81 const int64_t squared_prev_output = int64_t(uint64_t(prev_output) * uint64_t(prev_output)); in TEST()
124 const uint32_t prev_output = output - 1; in TEST() local
125 const int64_t squared_prev_output = int64_t(uint64_t(prev_output) * uint64_t(prev_output)); in TEST()
168 const uint32_t prev_output = output - 1; in TEST() local
169 const int64_t squared_prev_output = int64_t(uint64_t(prev_output) * uint64_t(prev_output)); in TEST()
212 const uint32_t prev_output = output - 1; in TEST() local
213 const int64_t squared_prev_output = int64_t(uint64_t(prev_output) * uint64_t(prev_output)); in TEST()
[all …]
/external/ComputeLibrary/src/runtime/CL/functions/
DCLArgMinMaxLayer.cpp136 …&compile_context, axis, op](const ICLTensor * input, const ICLTensor * prev_output, ICLTensor * ou… in configure()
139 …_reduction_kernels_vector.back()->configure(compile_context, input, prev_output, output, axis, op); in configure()
/external/pytorch/aten/src/ATen/core/
Dclass_type.cpp313 const Argument& prev_output = (hook_idx == 0) in checkForwardHookSchema() local
320 *prev_output.type() == *return_arg.type(), in checkForwardHookSchema()
325 prev_output.type()->annotation_str(), in checkForwardHookSchema()
/external/pytorch/torch/onnx/
Dsymbolic_opset9.py4337 prev_output = input
4424 inputs = [prev_output, weight_ih, weight_hh, bias_concat, sequence_lens]
4437 prev_output, h_out = g.op(
4446 prev_output, h_out = g.op(
4455 prev_output, h_out, c_out = g.op(
4467 prev_output = g.op("Transpose", prev_output, perm_i=[0, 2, 1, 3])
4468 prev_output = symbolic_helper._reshape_helper(
4470 prev_output,
4475 prev_output = symbolic_helper._squeeze_helper(g, prev_output, [1])
4482 prev_output = g.op("Transpose", prev_output, perm_i=[1, 0, 2])
[all …]
/external/tensorflow/tensorflow/python/keras/layers/
Drecurrent.py1377 prev_output = states[0] if nest.is_nested(states) else states
1380 prev_output, training)
1390 prev_output = prev_output * rec_dp_mask
1391 output = h + backend.dot(prev_output, self.recurrent_kernel)
/external/openthread/third_party/mbedtls/repo/library/
Daes.c1259 unsigned char *prev_output = output - 16; in mbedtls_aes_crypt_xts() local
1264 output[i] = prev_output[i]; in mbedtls_aes_crypt_xts()
1272 mbedtls_xor(tmp + i, prev_output + i, t + i, 16 - i); in mbedtls_aes_crypt_xts()
1281 mbedtls_xor(prev_output, tmp, t, 16); in mbedtls_aes_crypt_xts()
/external/mbedtls/library/
Daes.c1295 unsigned char *prev_output = output - 16; in mbedtls_aes_crypt_xts() local
1300 output[i] = prev_output[i]; in mbedtls_aes_crypt_xts()
1308 mbedtls_xor(tmp + i, prev_output + i, t + i, 16 - i); in mbedtls_aes_crypt_xts()
1317 mbedtls_xor(prev_output, tmp, t, 16); in mbedtls_aes_crypt_xts()
/external/tensorflow/tensorflow/python/keras/
Dbackend.py4327 prev_output = zeros_like(output)
4329 prev_output = successive_outputs[-1]
4331 output = array_ops.where_v2(tiled_mask_t, output, prev_output)
4459 def _step(time, output_ta_t, prev_output, *states): argument
4480 else nest.flatten(prev_output))
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision.cc2013 MutableGraphView::OutputPort prev_output = in NodeImplicitlyReadsNonResourceVariable() local
2015 const NodeDef* input = prev_output.node; in NodeImplicitlyReadsNonResourceVariable()
/external/pytorch/test/
Dtest_nn.py4365 prev_output = None
4378 if prev_output is not None:
4380 self.assertEqual(output1.data, prev_output)
4381 self.assertEqual(output2.data, prev_output)
4383 self.assertNotEqual(output1.data, prev_output)
4384 self.assertNotEqual(output2.data, prev_output)
4385 prev_output = output1.data