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

  /external/tensorflow/tensorflow/compiler/xla/tests/
batch_normalization_test.cc 503 *ReferenceUtil::Broadcast1DTo4D(mean, bounds, feature_index);
504 auto var4D = *ReferenceUtil::Broadcast1DTo4D(var, bounds, feature_index);
505 auto scale4D = *ReferenceUtil::Broadcast1DTo4D(scale, bounds, feature_index);
507 *ReferenceUtil::Broadcast1DTo4D(offset, bounds, feature_index);
604 *ReferenceUtil::Broadcast1DTo4D(mean, bounds, feature_index);
605 auto var4D = *ReferenceUtil::Broadcast1DTo4D(var, bounds, feature_index);
606 auto scale4D = *ReferenceUtil::Broadcast1DTo4D(scale, bounds, feature_index);
608 *ReferenceUtil::Broadcast1DTo4D(offset, bounds, feature_index);
723 *ReferenceUtil::Broadcast1DTo4D(mean, bounds, feature_index);
724 auto var4D = *ReferenceUtil::Broadcast1DTo4D(var, bounds, feature_index)
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/
reference_util.h 151 static std::unique_ptr<Array4D<float>> Broadcast1DTo4D(
reference_util.cc 647 /* static */ std::unique_ptr<Array4D<float>> ReferenceUtil::Broadcast1DTo4D(

Completed in 252 milliseconds