| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_subcomputation_unification_test.cc | 89 EXPECT_NE(x->to_apply(), y->to_apply());
102 EXPECT_EQ(x->to_apply(), y->to_apply());
128 EXPECT_NE(x->to_apply(), y->to_apply());
141 EXPECT_EQ(x->to_apply(), y->to_apply());
168 EXPECT_NE(x->to_apply(), y->to_apply());
[all...] |
| shape_inference.h | 86 const ProgramShape& to_apply,
136 const ProgramShape& to_apply);
245 // the to_apply parameters, and returns the to_apply result shape.
248 const ProgramShape& to_apply);
|
| flatten_call_graph.cc | 47 CHECK_EQ(instruction->to_apply(), computation);
|
| shape_inference_test.cc | 63 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_, f32_}, f32_); local
65 arg, f32_, dimensions_to_reduce, to_apply);
317 ProgramShape to_apply = ShapeUtil::MakeProgramShape( local
320 matrix_shape, init_value_shape, window, to_apply);
563 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_}, s32_); local
564 auto inferred_status = ShapeInference::InferMapShape({&arg}, to_apply, {0});
634 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_}, f32_); local
635 auto inferred_status = ShapeInference::InferMapShape({&arg}, to_apply, {0});
710 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_, f32_}, f32_); local
713 to_apply);
720 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_, f32_, f32_}, f32_); local
730 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_, f32_}, s32_); local
[all...] |
| liveness_util.cc | 255 auto* param = user->to_apply()->parameter_instruction(operand_indices[0]);
267 auto* callee_root = user->to_apply()->root_instruction();
366 auto* callee_root = user->to_apply()->root_instruction();
|
| service.cc | [all...] |
| hlo_cost_analysis.cc | 239 ProcessSubcomputation(map->to_apply()));
253 HloComputation* function = reduce->to_apply();
272 auto function = reduce_window->to_apply();
470 ProcessSubcomputation(call->to_apply()));
|
| inliner.cc | 67 HloComputation* function = map->to_apply();
|
| shape_inference.cc | [all...] |
| hlo_verifier.cc | 162 reduce->dimensions(), reduce->to_apply()->ComputeProgramShape()));
212 return CheckShape(call, call->to_apply()->ComputeProgramShape().result());
260 map->to_apply()->ComputeProgramShape(), map_dims));
269 reduce_window->to_apply()->ComputeProgramShape()));
[all...] |
| user_computation.cc | [all...] |
| hlo_instruction.cc | 1965 HloComputation* HloInstruction::to_apply() const { function in class:xla::HloInstruction
[all...] |
| hlo_ordering.cc | 182 call->to_apply())) {
|
| transpose_folding_test.cc | 158 HloComputation* callee_computation = call->to_apply();
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_parallelization_preparation.cc | 161 << " callee: " << call->to_apply()->root_instruction()->name();
162 call->to_apply()->root_instruction()->set_outer_dimension_partitions(
185 !instruction->to_apply()
|
| parallel_task_assignment.cc | 191 changed |= AssignParallelTasksHelper(module, instruction->to_apply(),
219 auto* new_root = call->to_apply()->root_instruction();
|
| elemental_ir_emitter.cc | 122 hlo->to_apply(), operands,
|
| parallel_cpu_executable.cc | 188 HloInstruction* root = instruction->to_apply()->root_instruction();
205 << instruction->to_apply()->root_instruction()->name()
224 << instruction->to_apply()->root_instruction()->name();
303 !instruction->to_apply()
|
| cpu_compiler.cc | 204 TF_RETURN_IF_ERROR(call->to_apply()->Accept(&candidates_for_call));
575 HloComputation* to_apply = instruction->to_apply(); local
576 parallel_computations.emplace(to_apply, instruction);
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| elemental_ir_emitter.cc | 326 TF_RET_CHECK(hlo->to_apply()->num_parameters() > 0);
333 return compute_nested_(*hlo->to_apply(), operand_elements);
408 compute_nested_(*hlo->to_apply(),
448 *hlo->to_apply(),
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| reduce_hlo_test.cc | 71 reduce = f32[2,2,3]{2,1,0} reduce(parameter, init_value), dimensions={1}, to_apply=Sum
|
| test_utils.cc | 166 op_num == 1 && LooksLikeSum(*instruction->to_apply())) ||
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| xla_data.proto | 539 ComputationHandle to_apply = 2;
582 ComputationHandle to_apply = 3;
596 // shape of to_apply.
603 ComputationHandle to_apply = 5;
610 ComputationHandle to_apply = 5;
|
| /external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
| hlo_parser_test.cc | 301 ROOT %call = f32[] call(f32[] %constant), to_apply=%Identity.v1
320 ROOT %reduce-window = f32[13,3,8,15]{0,3,2,1} reduce-window(f32[13,12,8,15]{0,3,2,1} %operand, f32[] %constant), window={size=1x1x7x1 stride=1x4x1x1 pad=0_0x0_0x3_3x0_0}, to_apply=%add_F32.v3
339 ROOT %reduce-window = f32[] reduce-window(f32[] %constant, f32[] %constant.1), to_apply=%add_F32.v3
742 ROOT map = f32[4]{0} map(param0, param1), to_apply=add_F32.v3
761 ROOT reduce = f32[8,16]{1,0} reduce(input, constant), dimensions={2}, to_apply=add_F32.v3
[all...] |
| hlo_parser.cc | 656 optional<HloComputation*> to_apply; local
657 attrs["to_apply"] = {/*required=*/true, AttrTy::kHloComputation,
658 &to_apply};
663 HloInstruction::CreateCall(shape, operands, *to_apply));
670 attrs["to_apply"] = {/*required=*/true, AttrTy::kHloComputation,
740 optional<HloComputation*> to_apply; local
741 attrs["to_apply"] = {/*required=*/true, AttrTy::kHloComputation,
742 &to_apply};
747 HloInstruction::CreateMap(shape, operands, *to_apply));
752 attrs["to_apply"] = {/*required=*/true, AttrTy::kHloComputation
[all...] |
