OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:kFeatureIndex
(Results
1 - 3
of
3
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_layout_assignment_test.cc
103
const int64
kFeatureIndex
= 1;
111
//
kFeatureIndex
dim of shape.
113
ShapeUtil::MakeShape(F32, {shape.dimensions(
kFeatureIndex
)});
137
Literal::CreateR0<int64>(
kFeatureIndex
)));
172
const int64
kFeatureIndex
= 1;
178
// rank 1, with as many elements are in the
kFeatureIndex
dim of shape.
180
ShapeUtil::MakeShape(F32, {shape.dimensions(
kFeatureIndex
)});
205
Literal::CreateR0<int64>(
kFeatureIndex
)));
242
const int64
kFeatureIndex
= 1;
248
// are rank 1, with as many elements are in the
kFeatureIndex
dim of shape
[
all
...]
/external/tensorflow/tensorflow/compiler/xla/tests/
bfloat16_test.cc
82
const int
kFeatureIndex
= 2;
98
/*epsilon=*/0.001,
kFeatureIndex
);
119
const int
kFeatureIndex
= 2;
141
/*epsilon=*/0.0,
kFeatureIndex
);
batch_normalization_test.cc
221
const int
kFeatureIndex
= 3;
232
/*epsilon=*/0.001,
kFeatureIndex
);
245
const int
kFeatureIndex
= 2;
256
/*epsilon=*/0.001,
kFeatureIndex
);
270
const int
kFeatureIndex
= 0;
287
/*epsilon=*/1,
kFeatureIndex
);
302
const int
kFeatureIndex
= 2;
320
/*epsilon=*/-100,
kFeatureIndex
);
334
const int
kFeatureIndex
= 2;
350
/*epsilon=*/0.0,
kFeatureIndex
);
[
all
...]
Completed in 276 milliseconds