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.cc503 *ReferenceUtil::Broadcast1DTo4D(mean, bounds, feature_index); in XLA_TEST_P()
504 auto var4D = *ReferenceUtil::Broadcast1DTo4D(var, bounds, feature_index); in XLA_TEST_P()
505 auto scale4D = *ReferenceUtil::Broadcast1DTo4D(scale, bounds, feature_index); in XLA_TEST_P()
507 *ReferenceUtil::Broadcast1DTo4D(offset, bounds, feature_index); in XLA_TEST_P()
604 *ReferenceUtil::Broadcast1DTo4D(mean, bounds, feature_index); in XLA_TEST_P()
605 auto var4D = *ReferenceUtil::Broadcast1DTo4D(var, bounds, feature_index); in XLA_TEST_P()
606 auto scale4D = *ReferenceUtil::Broadcast1DTo4D(scale, bounds, feature_index); in XLA_TEST_P()
608 *ReferenceUtil::Broadcast1DTo4D(offset, bounds, feature_index); in XLA_TEST_P()
723 *ReferenceUtil::Broadcast1DTo4D(mean, bounds, feature_index); in XLA_TEST_P()
724 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.cc647 /* static */ std::unique_ptr<Array4D<float>> ReferenceUtil::Broadcast1DTo4D( in Broadcast1DTo4D() function in xla::ReferenceUtil