Searched defs:mean_literal (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dbatch_normalization_test.cc605 auto mean_literal = Literal::CreateR1<float>(mean); local
615 auto mean_activations = builder.Parameter(3, mean_literal->shape(), "mean");
628 client_->TransferToServer(*mean_literal).ConsumeValueOrDie();
794 auto mean_literal = Literal::CreateR1<float>(mean); local
801 auto mean_parameter = builder.Parameter(2, mean_literal->shape(), "mean");
811 client_->TransferToServer(*mean_literal).ConsumeValueOrDie();

Completed in 121 milliseconds