Home
last modified time | relevance | path

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

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