| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_layout_assignment_test.cc | 272 Shape dot_shape = ShapeUtil::MakeShape(F32, {m, n}); local
277 HloInstruction::CreateParameter(1, dot_shape, "param1"));
281 builder.AddInstruction(CreateCanonicalDot(dot_shape, dot_lhs, dot_rhs));
285 dot_shape, HloOpcode::kAdd, dot_result, addend));
288 dot_shape, HloOpcode::kAdd, addend, dot_result));
295 dot_shape, HloInstruction::FusionKind::kOutput, add_result));
310 ShapeLayout(LayoutUtil::GetWithDefaultLayout(dot_shape));
312 ShapeLayout(LayoutUtil::GetWithDefaultLayout(dot_shape));
|
| cpu_instruction_fusion_test.cc | 638 Shape dot_shape = ShapeUtil::MakeShape(F32, {m, n}); local
645 HloInstruction::CreateParameter(2, dot_shape, "param2"));
648 builder.AddInstruction(CreateCanonicalDot(dot_shape, dot_lhs, dot_rhs));
650 HloInstruction::CreateBinary(dot_shape, HloOpcode::kAdd, dot, addend));
655 HloInstruction::CreateOutfeed(dot_shape, dot, token, "no_config"));
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| dot_decomposer.cc | 41 const Shape& dot_shape = dot->shape(); local
88 PrimitiveType dot_type = dot_shape.element_type();
89 const int64 dot_rows = dot_shape.dimensions(num_batch_dims + 0);
90 const int64 dot_cols = dot_shape.dimensions(num_batch_dims + 1);
155 HloInstruction::CreateReshape(dot_shape, concat));
|
| hlo_creation_utils.cc | 223 Shape dot_shape,
226 dot_shape, lhs, rhs, dim_numbers, precision_config));
|
| algebraic_simplifier_test.cc | 4335 Shape dot_shape = ShapeUtil::MakeShape(element_type, output_dims); local
4391 Shape dot_shape = ShapeUtil::MakeShape(element_type, {m, n}); local
4488 Shape dot_shape = ShapeUtil::MakeShape(F32, {spec.m, spec.n}); local
4553 Shape dot_shape = ShapeUtil::MakeShape(F32, {spec.m, spec.n}); local
4676 Shape dot_shape = ShapeUtil::MakeShape(F32, {dot_row_size, dot_col_size}); local
4749 Shape dot_shape = ShapeUtil::MakeShape(F32, {dot_row_size, dot_col_size}); local
[all...] |
| hlo_cost_analysis.cc | 241 const Shape& dot_shape = dot->shape(); local
251 kFmaFlops * ShapeUtil::ElementsIn(dot_shape) * reduction_width;
|
| hlo_evaluator.cc | 373 Shape dot_shape,
377 HloInstruction::CreateDot(dot_shape, lhs_instr.get(), rhs_instr.get(),
[all...] |
