OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:dot_shape
(Results
1 - 5
of
5
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/service/
dot_decomposer.cc
39
const Shape&
dot_shape
= dot->shape();
local
86
PrimitiveType dot_type =
dot_shape
.element_type();
87
const int64 dot_rows =
dot_shape
.dimensions(num_batch_dims + 0);
88
const int64 dot_cols =
dot_shape
.dimensions(num_batch_dims + 1);
152
HloInstruction::CreateReshape(
dot_shape
, concat));
algebraic_simplifier_test.cc
2582
Shape
dot_shape
= ShapeUtil::MakeShape(F32, {m, n});
local
2679
Shape
dot_shape
= ShapeUtil::MakeShape(F32, {spec.m, spec.n});
local
2743
Shape
dot_shape
= ShapeUtil::MakeShape(F32, {spec.m, spec.n});
local
[
all
...]
algebraic_simplifier.cc
284
const Shape&
dot_shape
, HloInstruction* lhs, int64 lhs_contracting_dim,
[
all
...]
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_layout_assignment_test.cc
265
Shape
dot_shape
= ShapeUtil::MakeShape(F32, {m, n});
local
270
HloInstruction::CreateParameter(1,
dot_shape
, "param1"));
274
HloInstruction::CreateCanonicalDot(
dot_shape
, dot_lhs, dot_rhs));
278
dot_shape
, HloOpcode::kAdd, dot_result, addend));
281
dot_shape
, HloOpcode::kAdd, addend, dot_result));
288
dot_shape
, HloInstruction::FusionKind::kOutput, add_result));
303
ShapeLayout(LayoutUtil::GetWithDefaultLayout(
dot_shape
));
305
ShapeLayout(LayoutUtil::GetWithDefaultLayout(
dot_shape
));
cpu_instruction_fusion_test.cc
629
Shape
dot_shape
= ShapeUtil::MakeShape(F32, {m, n});
local
636
HloInstruction::CreateParameter(2,
dot_shape
, "param2"));
639
HloInstruction::CreateCanonicalDot(
dot_shape
, dot_lhs, dot_rhs));
641
HloInstruction::CreateBinary(
dot_shape
, HloOpcode::kAdd, dot, addend));
645
HloInstruction::CreateOutfeed(
dot_shape
, dot, "no_config"));
Completed in 401 milliseconds