Searched defs:dot_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_layout_assignment_test.cc | 272 Shape dot_shape = ShapeUtil::MakeShape(F32, {m, n}); in RunDotOutputFusion() local
|
| D | cpu_instruction_fusion_test.cc | 640 Shape dot_shape = ShapeUtil::MakeShape(F32, {m, n}); in CreateComputationForDotAddOutputFusionTest() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_cost_analysis.cc | 325 const Shape& dot_shape = dot->shape(); in HandleDot() local
|
| D | algebraic_simplifier_test.cc | 5192 Shape dot_shape = ShapeUtil::MakeShape(element_type, output_dims); in TEST_P() local
5248 Shape dot_shape = ShapeUtil::MakeShape(element_type, {m, n}); in TEST_P() local
5352 Shape dot_shape = ShapeUtil::MakeShape(F32, {spec.m, spec.n}); in TEST_P() local
5417 Shape dot_shape = ShapeUtil::MakeShape(F32, {spec.m, spec.n}); in TEST_P() local
5636 Shape dot_shape = ShapeUtil::MakeShape(F32, {dot_row_size, dot_col_size}); in TEST_P() local
5709 Shape dot_shape = ShapeUtil::MakeShape(F32, {dot_row_size, dot_col_size}); in TEST_P() local
|