Searched refs:concat_size (Results 1 - 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_constant_folding_test.cc118 int64 concat_size = 0; local
122 concat_size += csize;
128 dimensions[test_config.concat_dimension] = concat_size;
/external/v8/src/
H A Delements.h137 uint32_t concat_size, uint32_t result_length);
H A Delements.cc3821 uint32_t concat_size,
3828 for (uint32_t i = 0; i < concat_size; i++) {
3856 for (uint32_t i = 0; i < concat_size; i++) {
3820 Concat(Isolate* isolate, Arguments* args, uint32_t concat_size, uint32_t result_len) argument
/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
H A Dpropagate_fixed_sizes.cc573 int concat_size = 0; local
584 concat_size += input_dims[axis];
586 // Write out the concat_size on the output array shape.
590 output_dims[axis] = concat_size;
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/
H A Dreference_ops.h1243 int concat_size = 0;
1250 concat_size += ArraySize(*input_dims[i], concat_dim);
1252 TFLITE_DCHECK_EQ(concat_size, ArraySize(output_dims, concat_dim));
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
H A Doptimized_ops.h2054 int concat_size = 0;
2061 concat_size += ArraySize(*input_dims[i], concat_dim);
2063 TFLITE_DCHECK_EQ(concat_size, ArraySize(output_dims, concat_dim));

Completed in 227 milliseconds