Home
last modified time | relevance | path

Searched refs:TfLiteIntArrayCopy (Results 1 – 25 of 86) sorted by relevance

1234

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dselect.cc105 output_size = TfLiteIntArrayCopy(input_x->dims); in SelectPrepare()
122 output_size = TfLiteIntArrayCopy(input_x->dims); in SelectPrepare()
H A Dwhile.cc77 TfLiteIntArrayCopy(src_tensor->dims))); in CopyTensorsShapeAndType()
164 dst_tensor->dims = TfLiteIntArrayCopy(src_tensor->dims); in ShallowCopyTensorsShapeTypeData()
329 TfLiteIntArray* output_size = TfLiteIntArrayCopy(body_output->dims); in Prepare_impl()
H A Dif.cc131 TfLiteIntArray* output_size = TfLiteIntArrayCopy(then_output->dims); in Prepare()
198 TfLiteIntArray* output_size = TfLiteIntArrayCopy(subgraph_output->dims); in Eval()
H A Dnumeric_verify.cc118 TfLiteIntArrayCopy(op_context.input->dims))); in Prepare()
125 TfLiteIntArrayCopy(op_context.input->dims)); in Prepare()
H A Dbidirectional_sequence_lstm.cc675 TfLiteIntArray* input_quantized_size = TfLiteIntArrayCopy(input->dims); in Prepare()
691 TfLiteIntArrayCopy(fw_activation_state->dims); in Prepare()
707 TfLiteIntArrayCopy(bw_activation_state->dims); in Prepare()
723 TfLiteIntArrayCopy(fw_cell_state->dims); in Prepare()
739 TfLiteIntArrayCopy(bw_cell_state->dims); in Prepare()
962 TfLiteIntArrayCopy(aux_input->dims); in Prepare()
H A Droll.cc133 TfLiteIntArray* cache_shape = TfLiteIntArrayCopy(input->dims); in AllocateTemporaryTensorsIfRequired()
182 TfLiteIntArray* output_shape = TfLiteIntArrayCopy(input->dims); in Prepare()
H A Dactivations.cc252 TfLiteIntArrayCopy(input->dims)); in GenericPrepare()
286 TfLiteIntArrayCopy(input->dims)); in ReluPrepare()
370 TfLiteIntArrayCopy(input->dims)); in LeakyReluPrepare()
465 TfLiteIntArrayCopy(input->dims)); in TanhPrepare()
568 TfLiteIntArrayCopy(input->dims)); in SigmoidPrepare()
657 TfLiteIntArrayCopy(input->dims)); in SoftmaxPrepare()
706 TfLiteIntArrayCopy(input->dims)); in LogSoftmaxPrepare()
H A Dbidirectional_sequence_rnn.cc208 TfLiteIntArray* input_quantized_size = TfLiteIntArrayCopy(input->dims); in Prepare()
224 TfLiteIntArrayCopy(fw_hidden_state->dims); in Prepare()
241 TfLiteIntArrayCopy(bw_hidden_state->dims); in Prepare()
340 TfLiteIntArrayCopy(aux_input->dims); in Prepare()
H A Dbasic_rnn.cc123 TfLiteIntArray* input_quantized_size = TfLiteIntArrayCopy(input->dims); in Prepare()
136 TfLiteIntArrayCopy(hidden_state->dims); in Prepare()
H A Dsvdf.cc179 TfLiteIntArray* input_quantized_size = TfLiteIntArrayCopy(input->dims); in Prepare()
211 TfLiteIntArrayCopy(weights_time->dims); in Prepare()
H A Dround.cc42 TfLiteIntArray* output_size = TfLiteIntArrayCopy(input->dims); in Prepare()
H A Dceil.cc41 TfLiteIntArray* output_size = TfLiteIntArrayCopy(input->dims); in Prepare()
H A Dzeros_like.cc43 TfLiteIntArrayCopy(input->dims)); in Prepare()
H A Dunidirectional_sequence_rnn.cc130 TfLiteIntArray* input_quantized_size = TfLiteIntArrayCopy(input->dims); in Prepare()
143 TfLiteIntArrayCopy(hidden_state->dims); in Prepare()
H A Dexp.cc45 TfLiteIntArray* output_dims = TfLiteIntArrayCopy(op_context.input->dims); in Prepare()
H A Dfake_quant.cc58 TfLiteIntArray* output_dims = TfLiteIntArrayCopy(op_context.input->dims); in Prepare()
H A Dneg.cc45 TfLiteIntArrayCopy(input->dims)); in Prepare()
H A Dfloor.cc47 TfLiteIntArray* output_size = TfLiteIntArrayCopy(input->dims); in Prepare()
H A Dread_variable.cc79 TfLiteIntArrayCopy(variable_tensor->dims))); in Eval()
H A Dsign.cc36 TfLiteIntArray* output_shape = TfLiteIntArrayCopy(input->dims); in PointwiseUnaryOpPrepare()
/aosp_15_r20/external/tensorflow/tensorflow/lite/c/
H A Dcommon_test.cc41 TfLiteIntArray* b = TfLiteIntArrayCopy(a); in TEST()
167 src.dims_signature = TfLiteIntArrayCopy(src.dims); in TEST()
H A Dcommon.cc68 TfLiteIntArray* TfLiteIntArrayCopy(const TfLiteIntArray* src) { in TfLiteIntArrayCopy() function
209 dst->dims = TfLiteIntArrayCopy(src->dims); in TfLiteTensorCopy()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/
H A Ddelegate_test_util.cc65 context, output, TfLiteIntArrayCopy(input1->dims))); in AddOpRegistration()
354 context, output, TfLiteIntArrayCopy(input1->dims))); in FakeFusedRegistration()
555 context, output, TfLiteIntArrayCopy(input->dims))); in FakeFusedRegistration()
/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dinterpreter_test.cc631 TfLiteIntArray* newSize = TfLiteIntArrayCopy(tensor0->dims); in GetPassthroughOpRegistration()
645 TfLiteIntArrayCopy(tensor0->dims)); in GetPassthroughOpRegistration()
865 TfLiteIntArray* newSize = TfLiteIntArrayCopy(tensor0->dims); in TEST()
1285 TfLiteIntArray* newSize = TfLiteIntArrayCopy(tensor0->dims); in CopyOpRegistration()
1483 TfLiteIntArray* new_size = TfLiteIntArrayCopy(in_tensor->dims); in CancelOpRegistration()
1505 TfLiteIntArray* new_size = TfLiteIntArrayCopy(in_tensor->dims); in OkOpRegistration()
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/resource/
H A Dresource_variable.cc68 tensor_.dims = TfLiteIntArrayCopy(tensor->dims); in AssignFrom()

1234