Searched refs:root_shape (Results 1 – 8 of 8) 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 …]
|
| D | conditional_canonicalizer.cc | 35 auto root_shape = conditional->shape(); in CanonicalizeNonTupleConditional() local
36 auto new_shape = ShapeUtil::MakeTupleShape({root_shape}); in CanonicalizeNonTupleConditional()
40 HloInstruction::CreateGetTupleElement(root_shape, new_conditional, 0)); in CanonicalizeNonTupleConditional()
|
| D | algebraic_simplifier_test.cc | 6526 const Shape root_shape = m->entry_computation()->root_instruction()->shape(); in TEST_F() local
6531 GmockMatch(m::Slice().WithShapeEqualTo(&root_shape))); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_nested.cc | 73 const Shape& root_shape = root->shape(); in CodegenNestedComputation() local
75 llvm_ir::ShapeToIrType(root_shape, module_)->getPointerTo()); in CodegenNestedComputation()
77 root_shape, ir_emitter_context_->llvm_module()->getDataLayout()); in CodegenNestedComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_executable.cc | 236 const Shape& root_shape = root->shape(); in CreateResultShapedBuffer() local
294 ShapeUtil::ByteSizeOf(ShapeUtil::GetSubshape(root_shape, index)); in CreateResultShapedBuffer()
|
| D | ir_emitter.cc | 1378 const Shape& root_shape = root_instruction->shape(); in MatchReductionGenerator() local
1379 if (ShapeUtil::ElementIsComplex(root_shape)) { in MatchReductionGenerator()
1387 bool root_is_floating_point = ShapeUtil::ElementIsFloating(root_shape); in MatchReductionGenerator()
1388 bool root_is_integral = ShapeUtil::ElementIsIntegral(root_shape); in MatchReductionGenerator()
1389 bool root_is_signed = ShapeUtil::ElementIsSigned(root_shape); in MatchReductionGenerator()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 582 const Shape& root_shape = in TEST_F() local
584 EXPECT_TRUE(root_shape.is_dynamic_dimension(0)); in TEST_F()
595 const Shape& root_shape = in TEST_F() local
598 EXPECT_FALSE(root_shape.is_dynamic_dimension(0)); in TEST_F()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 321 TensorShapeProto root_shape; member
389 group->root_shape = root_node_output_properties->shape(); in CreateOptimizedNodesGroup()
|