| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| broadcast_test.cc | 41 builder.AddInstruction(HloInstruction::CreateBroadcast(
57 builder.AddInstruction(HloInstruction::CreateBroadcast(
77 auto element1 = builder.AddInstruction(HloInstruction::CreateBroadcast(
79 auto element2 = builder.AddInstruction(HloInstruction::CreateBroadcast(
101 builder.AddInstruction(HloInstruction::CreateBroadcast(
120 builder.AddInstruction(HloInstruction::CreateBroadcast(
137 builder.AddInstruction(HloInstruction::CreateBroadcast(
157 builder.AddInstruction(HloInstruction::CreateBroadcast(
182 builder.AddInstruction(HloInstruction::CreateBroadcast(
213 builder.AddInstruction(HloInstruction::CreateBroadcast(
[all...] |
| multioutput_fusion_test.cc | 72 HloInstruction::CreateBroadcast(elem_shape2, add1, {}));
|
| fusion_test.cc | 277 builder.AddInstruction(HloInstruction::CreateBroadcast(shape, two, {}));
302 HloInstruction::CreateBroadcast(const_array->shape(), const_vector, {1}));
486 auto broadcast1 = builder.AddInstruction(HloInstruction::CreateBroadcast(
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| batchnorm_expander.cc | 196 HloInstruction::CreateBroadcast(operand_shape, scale, {feature_index}));
199 HloInstruction::CreateBroadcast(operand_shape, offset, {feature_index}));
236 HloInstruction::CreateBroadcast(operand_shape, mean, {feature_index}));
251 add(HloInstruction::CreateBroadcast(operand_shape, var, {feature_index}));
343 HloInstruction::CreateBroadcast(operand_shape, scale, {feature_index}));
346 HloInstruction::CreateBroadcast(operand_shape, offset, {feature_index}));
349 HloInstruction::CreateBroadcast(operand_shape, mean, {feature_index}));
352 add(HloInstruction::CreateBroadcast(operand_shape, var, {feature_index}));
469 auto scale_broadcasted = add(HloInstruction::CreateBroadcast(
471 auto variance_broadcasted = add(HloInstruction::CreateBroadcast(
[all...] |
| inliner.cc | 98 HloInstruction::CreateBroadcast(map->shape(), constant, {}));
|
| hlo_rematerialization_test.cc | 62 HloInstruction::CreateBroadcast(vec1024_shape_, param, {}));
105 HloInstruction::CreateBroadcast(vec1024_shape_, param, {}));
438 HloInstruction::CreateBroadcast(vec1024_shape_, param, {}));
554 HloInstruction::CreateBroadcast(vec1024_shape_, param, {}));
|
| liveness_util_test.cc | 288 HloInstruction::CreateBroadcast(data_shape, one, {1}));
326 HloInstruction::CreateBroadcast(data_shape, one, {1}));
355 HloInstruction::CreateBroadcast(data_shape, one, {1}));
434 HloInstruction::CreateBroadcast(shape, one, {1}));
|
| buffer_assignment_test.cc | [all...] |
| hlo_instruction_test.cc | [all...] |
| algebraic_simplifier.cc | 701 HloInstruction::CreateBroadcast(shape, hlo, {dim}));
[all...] |
| layout_assignment_test.cc | 409 HloInstruction::CreateBroadcast(bshape, param, {1, 2}));
452 HloInstruction::CreateBroadcast(f32_34, param, {3}));
458 HloInstruction::CreateBroadcast(f32_234, tanh, {2}));
581 HloInstruction::CreateBroadcast(input_shape, constant, {}));
[all...] |
| algebraic_simplifier_test.cc | 129 HloInstruction::CreateBroadcast(r2f32, zero, {0, 1}));
151 builder.AddInstruction(HloInstruction::CreateBroadcast(r2f32, zero, {1}));
[all...] |
| copy_insertion_test.cc | 448 HloInstruction::CreateBroadcast(data_shape_, induction_variable, {8}));
518 HloInstruction::CreateBroadcast(data_shape_, induction_variable, {8}));
677 HloInstruction::CreateBroadcast(data_shape_, one, {1}));
681 HloInstruction::CreateBroadcast(data_shape_, zero, {1}));
701 HloInstruction::CreateBroadcast(data_shape_, one, {1}));
714 HloInstruction::CreateBroadcast(data_shape_, one, {1}));
[all...] |
| hlo_cost_analysis_test.cc | 409 HloInstruction::CreateBroadcast(shape_without_layout, c2, {1}));
|
| hlo_instruction.cc | 664 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateBroadcast(
687 HloInstruction::CreateBroadcast(broadcast_shape, operand, {});
718 auto broadcast = HloInstruction::CreateBroadcast(
[all...] |
| hlo_instruction.h | 405 static std::unique_ptr<HloInstruction> CreateBroadcast(
[all...] |
| hlo_evaluator_test.cc | 347 b.AddInstruction(HloInstruction::CreateBroadcast(
365 b.AddInstruction(HloInstruction::CreateBroadcast(
[all...] |
| user_computation.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| instruction_fusion_test.cc | 36 builder.AddInstruction(HloInstruction::CreateBroadcast(
56 builder.AddInstruction(HloInstruction::CreateBroadcast(
|
| while_transformer_test.cc | 77 HloInstruction::CreateBroadcast(data_shape_, induction_variable, {}));
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_instruction_fusion_test.cc | 282 HloInstruction::CreateBroadcast(broadcast_shape, param0, {1}));
307 HloInstruction::CreateBroadcast(result_shape, param0, {1}));
|
| /external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
| hlo_parser.cc | 723 instruction = builder->AddInstruction(HloInstruction::CreateBroadcast(
[all...] |
