Home
last modified time | relevance | path

Searched refs:Broadcast1DTo4D (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dbatch_normalization_test.cc601 *ReferenceUtil::Broadcast1DTo4D(mean, bounds, feature_index); in XLA_TEST_P()
602 auto var4D = *ReferenceUtil::Broadcast1DTo4D(var, bounds, feature_index); in XLA_TEST_P()
603 auto scale4D = *ReferenceUtil::Broadcast1DTo4D(scale, bounds, feature_index); in XLA_TEST_P()
605 *ReferenceUtil::Broadcast1DTo4D(offset, bounds, feature_index); in XLA_TEST_P()
702 *ReferenceUtil::Broadcast1DTo4D(mean, bounds, feature_index); in XLA_TEST_P()
703 auto var4D = *ReferenceUtil::Broadcast1DTo4D(var, bounds, feature_index); in XLA_TEST_P()
704 auto scale4D = *ReferenceUtil::Broadcast1DTo4D(scale, bounds, feature_index); in XLA_TEST_P()
706 *ReferenceUtil::Broadcast1DTo4D(offset, bounds, feature_index); in XLA_TEST_P()
821 *ReferenceUtil::Broadcast1DTo4D(mean, bounds, feature_index); in XLA_TEST_P()
822 auto var4D = *ReferenceUtil::Broadcast1DTo4D(var, bounds, feature_index); in XLA_TEST_P()
[all …]
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.h151 static std::unique_ptr<Array4D<float>> Broadcast1DTo4D(
Dreference_util.cc582 /* static */ std::unique_ptr<Array4D<float>> ReferenceUtil::Broadcast1DTo4D( in Broadcast1DTo4D() function in xla::ReferenceUtil