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 500 auto normalized = *ReferenceUtil::BatchNorm4D(input_array, mean4D, var4D,
600 auto normalized = *ReferenceUtil::BatchNorm4D(input_array, mean4D, var4D,
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/
reference_util.h 211 static std::unique_ptr<Array4D<float>> BatchNorm4D(
reference_util.cc 408 /* static */ std::unique_ptr<Array4D<float>> ReferenceUtil::BatchNorm4D(

Completed in 78 milliseconds