Home
last modified time | relevance | path

Searched defs:concat_axis (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dconvert_trivial_tile_to_concat.cc54 int concat_axis = 0; in Run() local
/external/tensorflow/tensorflow/compiler/jit/
Dincrease_dynamism_for_auto_jit_pass.cc223 auto concat_axis = ops::Const(host_scope.WithOpName("concat_axis"), 0); in ComputeSliceSize() local
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_old_batch_norms_test.cc392 int32 concat_axis = split ? 3 : 2; in TestFoldFusedBatchNormsWithConcat() local