Home
last modified time | relevance | path

Searched refs:IsNullInput (Results 1 – 5 of 5) sorted by relevance

/frameworks/ml/nn/common/operations/
DLSTM.cpp49 return !IsNullInput(operand) ? reinterpret_cast<const T*>(operand->buffer) : nullptr; in GetOptionalBuffer()
158 if (!IsNullInput(input_to_input_weights)) { in CheckInputTensorDimensions()
172 if (!IsNullInput(recurrent_to_input_weights)) { in CheckInputTensorDimensions()
189 (!IsNullInput(input_to_input_weights) && !IsNullInput(recurrent_to_input_weights)) || in CheckInputTensorDimensions()
190 (IsNullInput(input_to_input_weights) && IsNullInput(recurrent_to_input_weights)); in CheckInputTensorDimensions()
193 if (!IsNullInput(cell_to_input_weights)) { in CheckInputTensorDimensions()
198 if (!IsNullInput(cell_to_forget_weights)) { in CheckInputTensorDimensions()
203 if (!IsNullInput(cell_to_output_weights)) { in CheckInputTensorDimensions()
209 params->use_cifg = IsNullInput(input_to_input_weights); in CheckInputTensorDimensions()
211 ((!IsNullInput(cell_to_input_weights) || params->use_cifg) && in CheckInputTensorDimensions()
[all …]
DBidirectionalSequenceLSTM.cpp49 return !IsNullInput(operand) ? reinterpret_cast<const T*>(operand->buffer) : nullptr; in GetOptionalBuffer()
190 params_.use_layer_norm = !IsNullInput(fw_input_layer_norm_weights_); in BidirectionalSequenceLSTM()
245 NN_RET_CHECK(!IsNullInput(GetInput(operation, operands, requiredInput))) in Prepare()
302 NN_RET_CHECK(IsNullInput(fw_aux_input_to_input_weights_) && in Prepare()
303 IsNullInput(bw_aux_input_to_input_weights_)); in Prepare()
307 ((params_.use_cifg || !IsNullInput(fw_aux_input_to_input_weights_)) && in Prepare()
308 !IsNullInput(fw_aux_input_to_forget_weights_) && in Prepare()
309 !IsNullInput(fw_aux_input_to_cell_weights_) && in Prepare()
310 !IsNullInput(fw_aux_input_to_output_weights_)) || in Prepare()
311 (IsNullInput(fw_aux_input_to_input_weights_) && in Prepare()
[all …]
DSVDF.cpp63 NN_RET_CHECK(!IsNullInput(GetInput(operation, operands, requiredInput))) in Prepare()
92 if (!IsNullInput(bias)) { in Prepare()
122 if (!IsNullInput(bias_)) { in Eval()
206 if (!IsNullInput(bias_)) { in EvalFloat32()
DLSHProjection.cpp50 NN_RET_CHECK(!IsNullInput(GetInput(operation, operands, requiredInput))) in Prepare()
/frameworks/ml/nn/common/include/
DCpuExecutor.h261 inline bool IsNullInput(const RunTimeOperandInfo* input) { in IsNullInput() function
269 [&operands](uint32_t i) { return !IsNullInput(&operands[i]); }); in NumInputsWithValues()