HomeSort by relevance Sort by last modified time
    Searched refs:BatchNorm4D (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/tests/
batch_normalization_test.cc 509 auto normalized = *ReferenceUtil::BatchNorm4D(input_array, mean4D, var4D,
610 auto normalized = *ReferenceUtil::BatchNorm4D(input_array, mean4D, var4D,
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/
reference_util.h 217 static std::unique_ptr<Array4D<float>> BatchNorm4D(
reference_util.cc 374 /* static */ std::unique_ptr<Array4D<float>> ReferenceUtil::BatchNorm4D(

Completed in 312 milliseconds