Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dconcatenation.h40 int64_t concat_size = 0; in Concatenation() local
91 int64_t concat_size = 0; in ConcatenationWithScaling() local
/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_gru.cc224 TfLiteIntArray* concat_size = TfLiteIntArrayCreate(2); in Prepare() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_constant_folding_test.cc126 int64_t concat_size = 0; in TEST_F() local
/external/pytorch/torch/csrc/jit/codegen/fuser/
Dexecutor.cpp300 std::vector<int64_t> concat_size(map_size.begin(), map_size.end()); in launchFusion() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc676 int concat_size = 0; in ProcessConcatenationOperator() local