Searched defs:dot_shape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dot_decomposer.cc | 41 const Shape& dot_shape = dot->shape(); in DecomposeBatchDot() local
|
| D | hlo_cost_analysis.cc | 241 const Shape& dot_shape = dot->shape(); in HandleDot() local
|
| D | algebraic_simplifier_test.cc | 4335 Shape dot_shape = ShapeUtil::MakeShape(element_type, output_dims); in TEST_P() local
4391 Shape dot_shape = ShapeUtil::MakeShape(element_type, {m, n}); in TEST_P() local
4488 Shape dot_shape = ShapeUtil::MakeShape(F32, {spec.m, spec.n}); in TEST_P() local
4553 Shape dot_shape = ShapeUtil::MakeShape(F32, {spec.m, spec.n}); in TEST_P() local
4676 Shape dot_shape = ShapeUtil::MakeShape(F32, {dot_row_size, dot_col_size}); in TEST_P() local
4749 Shape dot_shape = ShapeUtil::MakeShape(F32, {dot_row_size, dot_col_size}); in TEST_P() local
|
| /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 | 638 Shape dot_shape = ShapeUtil::MakeShape(F32, {m, n}); in CreateComputationForDotAddOutputFusionTest() local
|