/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
H A D | select.cc | 105 output_size = TfLiteIntArrayCopy(input_x->dims); in SelectPrepare() 122 output_size = TfLiteIntArrayCopy(input_x->dims); in SelectPrepare()
|
H A D | while.cc | 77 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 D | if.cc | 131 TfLiteIntArray* output_size = TfLiteIntArrayCopy(then_output->dims); in Prepare() 198 TfLiteIntArray* output_size = TfLiteIntArrayCopy(subgraph_output->dims); in Eval()
|
H A D | numeric_verify.cc | 118 TfLiteIntArrayCopy(op_context.input->dims))); in Prepare() 125 TfLiteIntArrayCopy(op_context.input->dims)); in Prepare()
|
H A D | bidirectional_sequence_lstm.cc | 675 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 D | roll.cc | 133 TfLiteIntArray* cache_shape = TfLiteIntArrayCopy(input->dims); in AllocateTemporaryTensorsIfRequired() 182 TfLiteIntArray* output_shape = TfLiteIntArrayCopy(input->dims); in Prepare()
|
H A D | activations.cc | 252 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 D | bidirectional_sequence_rnn.cc | 208 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 D | basic_rnn.cc | 123 TfLiteIntArray* input_quantized_size = TfLiteIntArrayCopy(input->dims); in Prepare() 136 TfLiteIntArrayCopy(hidden_state->dims); in Prepare()
|
H A D | svdf.cc | 179 TfLiteIntArray* input_quantized_size = TfLiteIntArrayCopy(input->dims); in Prepare() 211 TfLiteIntArrayCopy(weights_time->dims); in Prepare()
|
H A D | round.cc | 42 TfLiteIntArray* output_size = TfLiteIntArrayCopy(input->dims); in Prepare()
|
H A D | ceil.cc | 41 TfLiteIntArray* output_size = TfLiteIntArrayCopy(input->dims); in Prepare()
|
H A D | zeros_like.cc | 43 TfLiteIntArrayCopy(input->dims)); in Prepare()
|
H A D | unidirectional_sequence_rnn.cc | 130 TfLiteIntArray* input_quantized_size = TfLiteIntArrayCopy(input->dims); in Prepare() 143 TfLiteIntArrayCopy(hidden_state->dims); in Prepare()
|
H A D | exp.cc | 45 TfLiteIntArray* output_dims = TfLiteIntArrayCopy(op_context.input->dims); in Prepare()
|
H A D | fake_quant.cc | 58 TfLiteIntArray* output_dims = TfLiteIntArrayCopy(op_context.input->dims); in Prepare()
|
H A D | neg.cc | 45 TfLiteIntArrayCopy(input->dims)); in Prepare()
|
H A D | floor.cc | 47 TfLiteIntArray* output_size = TfLiteIntArrayCopy(input->dims); in Prepare()
|
H A D | read_variable.cc | 79 TfLiteIntArrayCopy(variable_tensor->dims))); in Eval()
|
H A D | sign.cc | 36 TfLiteIntArray* output_shape = TfLiteIntArrayCopy(input->dims); in PointwiseUnaryOpPrepare()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/c/ |
H A D | common_test.cc | 41 TfLiteIntArray* b = TfLiteIntArrayCopy(a); in TEST() 167 src.dims_signature = TfLiteIntArrayCopy(src.dims); in TEST()
|
H A D | common.cc | 68 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 D | delegate_test_util.cc | 65 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 D | interpreter_test.cc | 631 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 D | resource_variable.cc | 68 tensor_.dims = TfLiteIntArrayCopy(tensor->dims); in AssignFrom()
|