Searched refs:concat_size (Results 1 – 4 of 4) sorted by relevance
38 int64_t concat_size = 0; in Concatenation() local46 concat_size += input_shapes[i]->Dims(axis); in Concatenation()48 TFLITE_DCHECK_EQ(concat_size, output_shape.Dims(axis)); in Concatenation()89 int64_t concat_size = 0; in ConcatenationWithScaling() local97 concat_size += input_shapes[i]->Dims(axis); in ConcatenationWithScaling()99 TFLITE_DCHECK_EQ(concat_size, output_shape.Dims(axis)); in ConcatenationWithScaling()
216 TfLiteIntArray* concat_size = TfLiteIntArrayCreate(2); in Prepare() local217 concat_size->data[0] = n_batch; in Prepare()218 concat_size->data[1] = n_input + n_output; in Prepare()220 context->ResizeTensor(context, concat, concat_size)); in Prepare()
125 int64 concat_size = 0; in TEST_F() local129 concat_size += csize; in TEST_F()135 dimensions[test_config.concat_dimension] = concat_size; in TEST_F()
676 int concat_size = 0; in ProcessConcatenationOperator() local687 concat_size += input_dims[op->axis]; in ProcessConcatenationOperator()693 output_dims[op->axis] = concat_size; in ProcessConcatenationOperator()