Searched refs:concat_size (Results 1 – 5 of 5) sorted by relevance
123 int64 concat_size = 0; in TEST_F() local127 concat_size += csize; in TEST_F()133 dimensions[test_config.concat_dimension] = concat_size; in TEST_F()
127 uint32_t concat_size, uint32_t result_length);
4635 uint32_t concat_size, in Concat() argument4642 for (uint32_t i = 0; i < concat_size; i++) { in Concat()4670 for (uint32_t i = 0; i < concat_size; i++) { in Concat()
1612 int64_t concat_size = 0; in Concatenation() local1620 concat_size += input_shapes[i]->Dims(axis); in Concatenation()1622 TFLITE_DCHECK_EQ(concat_size, output_shape.Dims(axis)); in Concatenation()1665 int64_t concat_size = 0; in ConcatenationWithScaling() local1673 concat_size += input_shapes[i]->Dims(axis); in ConcatenationWithScaling()1675 TFLITE_DCHECK_EQ(concat_size, output_shape.Dims(axis)); in ConcatenationWithScaling()2236 int64_t concat_size = 0; in Split() local2244 concat_size += output_shapes[i]->Dims(axis); in Split()2246 TFLITE_DCHECK_EQ(concat_size, input_shape.Dims(axis)); in Split()
675 int concat_size = 0; in ProcessConcatenationOperator() local686 concat_size += input_dims[op->axis]; in ProcessConcatenationOperator()692 output_dims[op->axis] = concat_size; in ProcessConcatenationOperator()