Home
last modified time | relevance | path

Searched defs:concat_size (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dconcatenation.h38 int64_t concat_size = 0; in Concatenation() local
89 int64_t concat_size = 0; in ConcatenationWithScaling() local
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dunidirectional_sequence_gru.cc216 TfLiteIntArray* concat_size = TfLiteIntArrayCreate(2); in Prepare() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_constant_folding_test.cc125 int64 concat_size = 0; in TEST_F() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc676 int concat_size = 0; in ProcessConcatenationOperator() local