Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched refs:root_shape (Results 1 - 3 of 3) sorted by null

/external/tensorflow/tensorflow/compiler/xla/service/
reshape_mover_test.cc	41 auto root_shape = ShapeUtil::MakeShape(F32, {8, 7}); local 47 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, param0)); 49 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, param1)); 51 root_shape, HloOpcode::kAdd, reshape0, reshape1)); 78 auto root_shape = ShapeUtil::MakeShape(F32, {8, 7}); local 83 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, rng0)); 86 HloInstruction::CreateConstant(Literal::CreateFromShape(root_shape))); 89 root_shape, HloOpcode::kAdd, reshape0, const1)); 104 auto root_shape = ShapeUtil::MakeShape(F32, {}); local 110 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, param0)) 130 auto root_shape = ShapeUtil::MakeShape(F32, {8, 7}); local 177 auto root_shape = ShapeUtil::MakeShape(F32, {2, 3}); local 220 auto root_shape = ShapeUtil::MakeShape(F32, {8, 7}); local 256 auto root_shape = ShapeUtil::MakeShape(F32, {3, 2}); local 308 auto root_shape = ShapeUtil::MakeShape(F32, {2, 3}); local 347 auto root_shape = ShapeUtil::MakeShape(F32, {1, 1, 3}); local 375 auto root_shape = ShapeUtil::MakeShape(F32, {8, 7}); local 404 auto root_shape = ShapeUtil::MakeShape(F32, {8, 7}); local 437 auto root_shape = ShapeUtil::MakeShape(F32, {}); local     [all...]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
ir_emitter_nested.cc	64 const Shape& root_shape = root->shape(); local 66 llvm_ir::ShapeToIrType(root_shape, module_)->getPointerTo()); 68 root_shape, ir_emitter_context_->llvm_module()->getDataLayout());