Home
last modified time | relevance | path

Searched refs:TfLiteIntArrayEqual (Results 1 – 21 of 21) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/c/
H A Dcommon_test.cc62 ASSERT_FALSE(TfLiteIntArrayEqual(a, b)); in TEST()
63 ASSERT_TRUE(TfLiteIntArrayEqual(b, c)); in TEST()
64 ASSERT_TRUE(TfLiteIntArrayEqual(b, b)); in TEST()
65 ASSERT_FALSE(TfLiteIntArrayEqual(c, d)); in TEST()
176 EXPECT_EQ(1, TfLiteIntArrayEqual(dst.dims, src.dims)); in TEST()
H A Dcommon.cc41 int TfLiteIntArrayEqual(const TfLiteIntArray* a, const TfLiteIntArray* b) { in TfLiteIntArrayEqual() function
H A Dcommon.h110 int TfLiteIntArrayEqual(const TfLiteIntArray* a, const TfLiteIntArray* b);
/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dmodel_test.cc245 ASSERT_TRUE(TfLiteIntArrayEqual(node0.inputs, desired_inputs)); in TEST()
246 ASSERT_TRUE(TfLiteIntArrayEqual(node0.outputs, desired_outputs)); in TEST()
263 ASSERT_TRUE(TfLiteIntArrayEqual(node1.inputs, desired_inputs)); in TEST()
264 ASSERT_TRUE(TfLiteIntArrayEqual(node1.outputs, desired_outputs)); in TEST()
612 TfLiteIntArrayEqual(t1->sparsity->traversal_order, traversal_order)); in TEST()
618 ASSERT_TRUE(TfLiteIntArrayEqual(t1->sparsity->block_map, block_map)); in TEST()
634 ASSERT_TRUE(TfLiteIntArrayEqual(t1->sparsity->dim_metadata[1].array_segments, in TEST()
642 ASSERT_TRUE(TfLiteIntArrayEqual(t1->sparsity->dim_metadata[1].array_indices, in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dbidirectional_sequence_lstm.cc674 if (!TfLiteIntArrayEqual(input_quantized->dims, input->dims)) { in Prepare()
688 if (!TfLiteIntArrayEqual(fw_activation_state_quantized->dims, in Prepare()
704 if (!TfLiteIntArrayEqual(bw_activation_state_quantized->dims, in Prepare()
720 if (!TfLiteIntArrayEqual(fw_cell_state_quantized->dims, in Prepare()
736 if (!TfLiteIntArrayEqual(bw_cell_state_quantized->dims, in Prepare()
960 if (!TfLiteIntArrayEqual(aux_input_quantized->dims, aux_input->dims)) { in Prepare()
H A Dbidirectional_sequence_rnn.cc207 if (!TfLiteIntArrayEqual(input_quantized->dims, input->dims)) { in Prepare()
221 if (!TfLiteIntArrayEqual(fw_hidden_state_quantized->dims, in Prepare()
238 if (!TfLiteIntArrayEqual(bw_hidden_state_quantized->dims, in Prepare()
338 if (!TfLiteIntArrayEqual(aux_input_quantized->dims, aux_input->dims)) { in Prepare()
H A Dbasic_rnn.cc122 if (!TfLiteIntArrayEqual(input_quantized->dims, input->dims)) { in Prepare()
133 if (!TfLiteIntArrayEqual(hidden_state_quantized->dims, in Prepare()
H A Dsvdf.cc178 if (!TfLiteIntArrayEqual(input_quantized->dims, input->dims)) { in Prepare()
209 if (!TfLiteIntArrayEqual(float_weights_time->dims, weights_time->dims)) { in Prepare()
H A Dunidirectional_sequence_rnn.cc129 if (!TfLiteIntArrayEqual(input_quantized->dims, input->dims)) { in Prepare()
140 if (!TfLiteIntArrayEqual(hidden_state_quantized->dims, in Prepare()
H A Dwhile.cc161 if (!TfLiteIntArrayEqual(src_tensor->dims, dst_tensor->dims)) { in ShallowCopyTensorsShapeTypeData()
310 if (!TfLiteIntArrayEqual(body_input->dims, body_output->dims)) { in Prepare_impl()
H A Dif.cc114 if (!TfLiteIntArrayEqual(then_output->dims, else_output->dims)) { in Prepare()
H A Dunidirectional_sequence_lstm.cc997 if (!TfLiteIntArrayEqual(input_quantized->dims, input->dims)) { in Prepare()
1010 if (!TfLiteIntArrayEqual(output_state_quantized->dims, in Prepare()
1026 if (!TfLiteIntArrayEqual(cell_state_quantized->dims, cell_state->dims)) { in Prepare()
H A Dkernel_util.cc410 return TfLiteIntArrayEqual(input1->dims, input2->dims); in HaveSameShapes()
H A Dlstm.cc1447 if (!TfLiteIntArrayEqual(input_quantized->dims, input->dims)) { in Prepare()
1460 if (!TfLiteIntArrayEqual(output_state_quantized->dims, in Prepare()
1476 if (!TfLiteIntArrayEqual(cell_state_quantized->dims, cell_state->dims)) { in Prepare()
H A Ddepthwise_conv.cc242 if (!TfLiteIntArrayEqual(input_quantized->dims, input->dims)) { in Prepare()
H A Dconv.cc557 if (!TfLiteIntArrayEqual(input_quantized->dims, input->dims)) { in Prepare()
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/resource/
H A Dresource_variable.cc64 if (TfLiteIntArrayEqual(old_dims, tensor->dims)) { in AssignFrom()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/
H A Dserialization_test.cc342 ASSERT_EQ(TfLiteIntArrayEqual(nodes_to_delegate_array, read_back_array), 1); in TEST_F()
/aosp_15_r20/external/mesa3d/include/tensorflow/lite/core/c/
H A Dcommon.h119 int TfLiteIntArrayEqual(const TfLiteIntArray* a, const TfLiteIntArray* b);
/aosp_15_r20/external/tensorflow/tensorflow/lite/core/
H A Dsubgraph.cc1574 TfLiteIntArrayEqual(tensor->dims, new_size) == 0; in ResizeTensorImpl()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate.cc3220 Expect(TfLiteIntArrayEqual(condition_shape, input_shape), in Validate()