Searched refs:concat_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_concat_op.cc | 147 auto concat_shape = concat.shape(); in Compute() local
149 output_shape(j) = concat_shape[j]; in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils.cc | 211 TF_ASSIGN_OR_RETURN(Shape concat_shape, ShapeInference::InferConcatOpShape( in MakeConcatHlo()
214 HloInstruction::CreateConcatenate(concat_shape, operands, dimension)); in MakeConcatHlo()
|
| D | buffer_assignment_test.cc | 1703 const Shape concat_shape = ShapeUtil::MakeShape(F32, {200}); in TEST_F() local
1705 HloInstruction::CreateConcatenate(concat_shape, {rev, neg}, 0)); in TEST_F()
|
| D | algebraic_simplifier_test.cc | 1719 Shape concat_shape = in TEST_F() local
1723 concat_shape, {param0, param1, param2}, 1)); in TEST_F()
1811 Shape concat_shape = ShapeUtil::MakeShape(F32, {50, 80}); in TEST_F() local
1860 concat_shape, in TEST_F()
|