Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dsparse_concat_op.cc147 auto concat_shape = concat.shape(); in Compute() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment_test.cc1703 const Shape concat_shape = ShapeUtil::MakeShape(F32, {200}); in TEST_F() local
Dalgebraic_simplifier_test.cc1719 Shape concat_shape = in TEST_F() local
1811 Shape concat_shape = ShapeUtil::MakeShape(F32, {50, 80}); in TEST_F() local