Home
last modified time | relevance | path

Searched refs:concat_size (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_constant_folding_test.cc123 int64 concat_size = 0; in TEST_F() local
127 concat_size += csize; in TEST_F()
133 dimensions[test_config.concat_dimension] = concat_size; in TEST_F()
/external/v8/src/
Delements.h127 uint32_t concat_size, uint32_t result_length);
Delements.cc4635 uint32_t concat_size, in Concat() argument
4642 for (uint32_t i = 0; i < concat_size; i++) { in Concat()
4670 for (uint32_t i = 0; i < concat_size; i++) { in Concat()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h1612 int64_t concat_size = 0; in Concatenation() local
1620 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() local
1673 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() local
2244 concat_size += output_shapes[i]->Dims(axis); in Split()
2246 TFLITE_DCHECK_EQ(concat_size, input_shape.Dims(axis)); in Split()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc675 int concat_size = 0; in ProcessConcatenationOperator() local
686 concat_size += input_dims[op->axis]; in ProcessConcatenationOperator()
692 output_dims[op->axis] = concat_size; in ProcessConcatenationOperator()