Searched refs:root_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | reshape_mover_test.cc | 42 auto root_shape = ShapeUtil::MakeShape(F32, {8, 7}); in TEST_F() local
48 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, param0)); in TEST_F()
50 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, param1)); in TEST_F()
52 root_shape, HloOpcode::kAdd, reshape0, reshape1)); in TEST_F()
80 auto root_shape = ShapeUtil::MakeShape(F32, {8, 7}); variable
89 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, rng0));
92 HloInstruction::CreateConstant(Literal::CreateFromShape(root_shape)));
95 root_shape, HloOpcode::kAdd, reshape0, const1));
111 auto root_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
117 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, param0)); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_nested.cc | 65 const Shape& root_shape = root->shape(); in EmitBasePointersForNestedComputation() local
67 llvm_ir::ShapeToIrType(root_shape, module_)->getPointerTo()); in EmitBasePointersForNestedComputation()
69 root_shape, ir_emitter_context_->llvm_module()->getDataLayout()); in EmitBasePointersForNestedComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 1413 const Shape& root_shape = root_instruction->shape(); in MatchReductionGenerator() local
1414 if (ShapeUtil::ElementIsComplex(root_shape)) { in MatchReductionGenerator()
1422 bool root_is_floating_point = ShapeUtil::ElementIsFloating(root_shape); in MatchReductionGenerator()
1423 bool root_is_integral = ShapeUtil::ElementIsIntegral(root_shape); in MatchReductionGenerator()
1424 bool root_is_signed = ShapeUtil::ElementIsSigned(root_shape); in MatchReductionGenerator()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 296 TensorShapeProto root_shape; member
364 group->root_shape = root_node_output_properties.shape(); in CreateOptimizedNodesGroup()
|