Searched refs:offset_literal (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dbatch_normalization_test.cc505 auto offset_literal = Literal::CreateR1<float>(offset); local
514 builder.Parameter(2, offset_literal->shape(), "scale");
525 client_->TransferToServer(*offset_literal).ConsumeValueOrDie();
603 auto offset_literal = Literal::CreateR1<float>(offset); local
614 builder.Parameter(2, offset_literal->shape(), "scale");
626 client_->TransferToServer(*offset_literal).ConsumeValueOrDie();

Completed in 129 milliseconds