Cross Reference: /external/tensorflow/tensorflow/compiler/xla/tests/local_client_execute

Lines Matching refs:ShapedBufferToLiteral
65   LiteralTestUtil::ExpectR0Near<float>(123.f, *ShapedBufferToLiteral(*result),
79   LiteralTestUtil::ExpectR0Near<float>(165.f, *ShapedBufferToLiteral(*result),
93   LiteralTestUtil::ExpectR1Near<float>({}, *ShapedBufferToLiteral(*result),
109       {2.0f, 4.0f, 6.0f}, *ShapedBufferToLiteral(*result), error_spec_);
127       {2.0f, 4.0f, 6.0f}, *ShapedBufferToLiteral(*result), error_spec_);
153                                        *ShapedBufferToLiteral(*result_colmaj),
161       *ShapedBufferToLiteral(*result_param_swap), error_spec_);
185                                        *ShapedBufferToLiteral(*result_colmaj),
197                                        *ShapedBufferToLiteral(*result_rowmaj),
219   std::unique_ptr<Literal> result_literal = ShapedBufferToLiteral(*result);
248   std::unique_ptr<Literal> result_literal = ShapedBufferToLiteral(*result);
283   std::unique_ptr<Literal> result_literal = ShapedBufferToLiteral(*result);
329   std::unique_ptr<Literal> result_literal = ShapedBufferToLiteral(*result);
371   std::unique_ptr<Literal> result_literal = ShapedBufferToLiteral(*result);
401   std::unique_ptr<Literal> result_0_literal = ShapedBufferToLiteral(*result_0);
410   std::unique_ptr<Literal> result_1_literal = ShapedBufferToLiteral(*result_1);
459   std::unique_ptr<Literal> result_literal = ShapedBufferToLiteral(*result);
516   std::unique_ptr<Literal> result_literal = ShapedBufferToLiteral(*result);
561   std::unique_ptr<Literal> result_literal = ShapedBufferToLiteral(*result);
649                                             *ShapedBufferToLiteral(*result));
694                                           *ShapedBufferToLiteral(*result));
772   std::unique_ptr<Literal> tuple_literal = ShapedBufferToLiteral(*result);
801       {2.0f, 4.0f, 6.0f}, *ShapedBufferToLiteral(*result), error_spec_);
814         local_client_->ShapedBufferToLiteral(*shaped_buffer));
854         local_client_->ShapedBufferToLiteral(*shaped_buffer));