Searched refs:CreateConstantFromLiteral (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reduce_window_test.cc | 76 auto init = CreateConstantFromLiteral(LiteralUtil::CreateR0<float>(0.0f), in ReduceWindowAdd()
88 CreateConstantFromLiteral(LiteralUtil::MinValue(F32), &builder_); in ReduceWindowMax()
99 CreateConstantFromLiteral(LiteralUtil::MaxValue(F32), &builder_); in ReduceWindowMin()
109 const auto input = CreateConstantFromLiteral( in TEST_P()
112 CreateConstantFromLiteral(LiteralUtil::CreateR0<float>(0), &builder_); in TEST_P()
127 CreateConstantFromLiteral(LiteralUtil::CreateR0<float>(42.0), &builder_); in TEST_P()
129 CreateConstantFromLiteral(LiteralUtil::CreateR0<float>(1.0), &builder_); in TEST_P()
138 const auto input = CreateConstantFromLiteral( in TEST_P()
146 const auto input = CreateConstantFromLiteral( in TEST_P()
316 CreateConstantFromLiteral(LiteralUtil::CreateR0<float>(8.0f), b.get())); in XLA_TEST_P()
[all …]
|
| D | client_library_test_base.h | 298 XlaOp CreateConstantFromLiteral(const Literal& literal, XlaBuilder* builder);
307 return CreateConstantFromLiteral(LiteralUtil::CreateFromArray(array), in CreateConstantFromArray()
314 return CreateConstantFromLiteral(LiteralUtil::CreateR0<NativeT>(value), in CreateConstantFromScalar()
|
| D | client_library_test_base.cc | 601 XlaOp ClientLibraryTestBase::CreateConstantFromLiteral(const Literal& literal, in CreateConstantFromLiteral() function in xla::ClientLibraryTestBase
|