OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:batchnorm
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_layout_assignment_test.cc
139
auto*
batchnorm
= builder.AddInstruction(HloInstruction::CreateCustomCall(
local
146
module->AddEntryComputation(builder.Build(
batchnorm
));
162
// The first operand to
batchnorm
should have the same layout as the
164
EXPECT_TRUE(LayoutUtil::Equal(
batchnorm
->operand(0)->shape().layout(),
165
batchnorm
->shape().layout()))
166
<<
batchnorm
->ToString();
207
auto*
batchnorm
= builder.AddInstruction(HloInstruction::CreateCustomCall(
local
213
module->AddEntryComputation(builder.Build(
batchnorm
));
231
// The first operand to
batchnorm
should have the same layout as the
234
LayoutUtil::Equal(
batchnorm
->operand(0)->shape().layout()
283
auto*
batchnorm
=
local
[
all
...]
/external/tensorflow/tensorflow/core/graph/
graph_constructor_test.cc
1099
name: "
batchnorm
"
[
all
...]
Completed in 212 milliseconds