HomeSort by relevance Sort by last modified time
    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