Home
last modified time | relevance | path

Searched refs:CreateFromArray (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dreshape_test.cc239 auto input_literal = LiteralUtil::CreateFromArray(Array2D<float>(0, 2)); in XLA_TEST_P()
255 auto input_literal = LiteralUtil::CreateFromArray(*simple); in XLA_TEST_P()
264 auto expected_literal = LiteralUtil::CreateFromArray(*expected); in XLA_TEST_P()
273 auto input_literal = LiteralUtil::CreateFromArray(*a4x3); in XLA_TEST_P()
282 auto expected_literal = LiteralUtil::CreateFromArray(*expected); in XLA_TEST_P()
290 auto input_literal = LiteralUtil::CreateFromArray(Array2D<float>(0, 4)); in XLA_TEST_P()
305 auto input_literal = LiteralUtil::CreateFromArray(*a4x3); in XLA_TEST_P()
313 auto expected_literal = LiteralUtil::CreateFromArray(*expected); in XLA_TEST_P()
322 auto input_literal = LiteralUtil::CreateFromArray(Array2D<float>(6, 0)); in XLA_TEST_P()
330 LiteralUtil::CreateFromArray(Array4D<float>(2, 3, 0, 0)); in XLA_TEST_P()
[all …]
Dreduce_window_test.cc173 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*res), {}, in XLA_TEST_P()
188 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*res), {}, in XLA_TEST_P()
202 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*res), {}, in XLA_TEST_P()
219 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*res), {}, in XLA_TEST_P()
241 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*result), {}, in XLA_TEST_P()
278 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*result), {}, in XLA_TEST_P()
303 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*result), {}, in XLA_TEST_P()
320 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*result), {}, in XLA_TEST_P()
358 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*expected), in XLA_TEST_P()
378 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*res), in XLA_TEST_P()
[all …]
Dconvolution_test.cc133 {LiteralUtil::CreateFromArray(input_data), in RunTest()
134 LiteralUtil::CreateFromArray(filter_data)}, in RunTest()
167 {LiteralUtil::CreateFromArray(input_data), in RunTest()
168 LiteralUtil::CreateFromArray(filter_data)}, in RunTest()
202 {LiteralUtil::CreateFromArray(input_data), in RunTest()
203 LiteralUtil::CreateFromArray(filter_data)}, in RunTest()
234 {LiteralUtil::CreateFromArray(input_data), in RunTest()
235 LiteralUtil::CreateFromArray(filter_data)}, in RunTest()
1540 {LiteralUtil::CreateFromArray(param0), in XLA_TEST_P()
1541 LiteralUtil::CreateFromArray(param1)}, in XLA_TEST_P()
[all …]
Dclient_library_test_base.h300 return AddParam(LiteralUtil::CreateFromArray(argument), builder); in AddParam()
315 return CreateConstantFromLiteral(LiteralUtil::CreateFromArray(array), in CreateConstantFromArray()
583 Literal expected_literal = LiteralUtil::CreateFromArray<NativeT>(expected); in ComputeAndCompare()
599 Literal expected_literal = LiteralUtil::CreateFromArray<NativeT>(expected); in ComputeAndCompare()
678 Literal literal = LiteralUtil::CreateFromArray(array); in CreateParameter()
Dbatch_normalization_test.cc279 {LiteralUtil::CreateFromArray(out), LiteralUtil::CreateR1<float>({4, 5}), in XLA_TEST_P()
333 {LiteralUtil::CreateFromArray(out), LiteralUtil::CreateR1<float>({4, 5}), in XLA_TEST_P()
463 {LiteralUtil::CreateFromArray(out), LiteralUtil::CreateR1<float>({0, 0}), in XLA_TEST_P()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.h127 static Literal CreateFromArray(const Array<NativeT>& values);
413 /* static */ Literal LiteralUtil::CreateFromArray( in CreateFromArray() function
428 return CreateFromArray(values); in CreateR2FromArray2D()
440 return CreateFromArray(values); in CreateR3FromArray3D()
499 return CreateFromArray(values); in CreateR4FromArray4D()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dsorting_test.cc126 CreateConstantFromLiteral(LiteralUtil::CreateFromArray(input), &builder); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_propagation_test.cc241 HloInstruction::CreateConstant(LiteralUtil::CreateFromArray(array_a))); in TEST_F()
243 HloInstruction::CreateConstant(LiteralUtil::CreateFromArray(array_b))); in TEST_F()
257 LiteralUtil::ConvertF32ToBF16(LiteralUtil::CreateFromArray(array_a)), in TEST_F()
260 LiteralUtil::ConvertF32ToBF16(LiteralUtil::CreateFromArray(array_b)), in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h2596 LiteralUtil::CreateFromArray<NativeT>(values)); in ConstantFromArray()