Home
last modified time | relevance | path

Searched refs:output_state (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dlstm_test.cc44 TensorRef<BHWC> output_state; in TEST() local
45 output_state.type = DataType::FLOAT32; in TEST()
46 output_state.ref = 2; in TEST()
47 output_state.shape = BHWC(1, 2, 2, 1); in TEST()
58 {input, prev_state}, {output_state, output_activation}); in TEST()
/external/webrtc/sdk/android/native_api/stacktrace/
Dstacktrace.cc105 SignalHandlerOutputState* const output_state = in UnwindBacktrace() local
109 if (output_state == nullptr) in UnwindBacktrace()
113 if (output_state->stack_size_counter >= kMaxStackSize) in UnwindBacktrace()
119 output_state->addresses[output_state->stack_size_counter] = in UnwindBacktrace()
121 ++output_state->stack_size_counter; in UnwindBacktrace()
/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.cc153 const float* output_state, const float* recurrent_to_gate_weights, in CalculateLstmGateFloat() argument
185 recurrent_to_gate_weights, n_cell, n_output, output_state, n_batch, gate); in CalculateLstmGateFloat()
267 const float proj_clip, float* output_state, in CalculateLstmOutputFloat() argument
280 output_state); in CalculateLstmOutputFloat()
282 std::fill_n(output_state, n_batch * n_output, 0.0f); in CalculateLstmOutputFloat()
285 projection_weights, n_output, n_cell, scratch, n_batch, output_state); in CalculateLstmOutputFloat()
287 tensor_utils::CwiseClipping(output_state, n_batch * n_output, proj_clip); in CalculateLstmOutputFloat()
290 std::copy_n(scratch, n_batch * n_output, output_state); in CalculateLstmOutputFloat()
310 const int8_t* output_state, const float* output_state_sf, in CalculateLstmGateHybrid() argument
384 n_output, output_state, scales.data(), n_batch, gate); in CalculateLstmGateHybrid()
[all …]
Dlstm_eval.h123 int output_offset, TfLiteTensor* scratch_buffer, TfLiteTensor* output_state,
166 TfLiteTensor* output_state, TfLiteTensor* cell_state,
193 TfLiteTensor* output_state, TfLiteTensor* cell_state, TfLiteTensor* output,
217 const TfLiteLSTMParams* params, TfLiteTensor* output_state,
Dunidirectional_sequence_lstm.cc145 TfLiteTensor* output_state = in PopulateQuantizedLstmParams8x8_16() local
147 TF_LITE_ENSURE(context, output_state != nullptr); in PopulateQuantizedLstmParams8x8_16()
245 output_state_scale = output_state->params.scale; in PopulateQuantizedLstmParams8x8_16()
711 const TfLiteTensor* output_state = in PopulatePrecomputedZPTimesWeightsWithBias() local
713 TF_LITE_ENSURE(context, output_state != nullptr); in PopulatePrecomputedZPTimesWeightsWithBias()
716 const int32_t output_state_zero_point = -output_state->params.zero_point; in PopulatePrecomputedZPTimesWeightsWithBias()
917 TfLiteTensor* output_state = in Prepare() local
919 TF_LITE_ENSURE(context, output_state != nullptr); in Prepare()
927 TF_LITE_ENSURE_EQ(context, NumElements(output_state), n_batch * n_output); in Prepare()
999 output_state->dims)) { in Prepare()
[all …]
Dlstm.cc230 TfLiteTensor* output_state = in PopulateQuantizedLstmParams8x8_16() local
232 TF_LITE_ENSURE(context, output_state != nullptr); in PopulateQuantizedLstmParams8x8_16()
330 output_state_scale = output_state->params.scale; in PopulateQuantizedLstmParams8x8_16()
532 TfLiteTensor* output_state = in PopulateQuantizedLstmParams8x8_8() local
534 TF_LITE_ENSURE(context, output_state != nullptr); in PopulateQuantizedLstmParams8x8_8()
649 output_state_scale = output_state->params.scale; in PopulateQuantizedLstmParams8x8_8()
666 output_state_ptr = output_state->data.int8; in PopulateQuantizedLstmParams8x8_8()
670 output_state_zp = output_state->params.zero_point; in PopulateQuantizedLstmParams8x8_8()
1157 const TfLiteTensor* output_state = in PopulatePrecomputedZPTimesWeightsWithBias() local
1159 TF_LITE_ENSURE(context, output_state != nullptr); in PopulatePrecomputedZPTimesWeightsWithBias()
[all …]
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dunidirectional_sequence_gru.cc34 TfLiteTensor* output_state, TfLiteTensor* activation, in GruImpl() argument
58 float* output_state_data = GetTensorData<float>(output_state); in GruImpl()
187 TfLiteTensor* output_state; in Prepare() local
189 GetOutputSafe(context, node, kOutputState, &output_state)); in Prepare()
191 context, context->ResizeTensor(context, output_state, in Prepare()
245 TfLiteTensor* output_state; in Eval() local
247 GetOutputSafe(context, node, kOutputState, &output_state)); in Eval()
257 candidate_bias, output, output_state, activation, concat, in Eval()
Dgru_cell.cc40 float* output, float* output_state, in GruCell() argument
88 memcpy(output_state, output, n_batch * n_output * sizeof(float)); in GruCell()
Dgru_cell.h35 float* output, float* output_state,
/external/tensorflow/tensorflow/compiler/xla/service/
Drng_bit_generator_expander.cc42 XlaOp GetPhiloxOutputStateOp(XlaOp output_state, const Shape& state_shape) { in GetPhiloxOutputStateOp() argument
44 output_state = Slice(output_state, {0}, {1}, {1}); in GetPhiloxOutputStateOp()
46 return output_state; in GetPhiloxOutputStateOp()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc43 const float* output_state, const float* recurrent_to_gate_weights, in CalculateLstmGateFloat() argument
77 recurrent_to_gate_weights, n_cell, n_output, output_state, n_batch, gate); in CalculateLstmGateFloat()
127 const float proj_clip, float* output_state, float* scratch, Logger* logger, in CalculateLstmOutputCalibration() argument
143 output_state); in CalculateLstmOutputCalibration()
145 std::fill_n(output_state, n_batch * n_output, 0.0f); in CalculateLstmOutputCalibration()
148 projection_weights, n_output, n_cell, scratch, n_batch, output_state); in CalculateLstmOutputCalibration()
150 tensor_utils::CwiseClipping(output_state, n_batch * n_output, proj_clip); in CalculateLstmOutputCalibration()
153 std::copy_n(scratch, n_batch * n_output, output_state); in CalculateLstmOutputCalibration()
288 int output_offset, TfLiteTensor* scratch_buffer, TfLiteTensor* output_state, in EvalCalibration()
375 GetTensorData<float>(output_state), GetTensorData<float>(cell_state), in EvalCalibration()
[all …]
/external/arm-trusted-firmware/plat/arm/css/common/
Dcss_pm.c329 psci_power_state_t *output_state) in css_translate_power_state_by_mpidr() argument
331 return arm_validate_power_state(power_state, output_state); in css_translate_power_state_by_mpidr()
/external/arm-trusted-firmware/plat/arm/board/fvp/
Dfvp_pm.c411 psci_power_state_t *output_state) in fvp_translate_power_state_by_mpidr() argument
413 return arm_validate_power_state(power_state, output_state); in fvp_translate_power_state_by_mpidr()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dlstm_parser.cc116 Value* output_state, Value* cell_state, in BuildLstmGate() argument
150 RETURN_IF_ERROR(graph->AddConsumer(node->id, output_state->id)); in BuildLstmGate()
337 Value* output_state, Value* output_gate, in BuildOutputStateUpdate() argument
376 Value* projected_output_state = CreateNewSimilarValue(graph, output_state); in BuildOutputStateUpdate()
/external/tensorflow/tensorflow/python/autograph/operators/
Dcontrol_flow_deprecated_py2.py713 output_aug_vars, output_state, extra_cond = scan_outputs
715 return output_aug_vars, output_state
781 output_aug_vars, output_state = scan_outputs
782 return output_aug_vars, output_state
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler.cc850 auto& output_state = node_map_[output_node]; in GetOutputNodes() local
851 output_state.num_inputs_ready++; in GetOutputNodes()
855 int output_state_inputs_size = output_state.inputs.size(); in GetOutputNodes()
856 if (output_state.num_inputs_ready == output_state_inputs_size || in GetOutputNodes()
859 output_state.time_ready = curr_time; in GetOutputNodes()
/external/arm-trusted-firmware/include/lib/psci/
Dpsci.h321 psci_power_state_t *output_state);
/external/arm-trusted-firmware/docs/getting_started/
Dporting-guide.rst2440 populated in the ``output_state`` (third argument) array. The functionality