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.cc227 auto input_literal = LiteralUtil::CreateFromArray(Array2D<float>(0, 2)); in XLA_TEST_P()
242 auto input_literal = LiteralUtil::CreateFromArray(*simple); in XLA_TEST_P()
250 auto expected_literal = LiteralUtil::CreateFromArray(*expected); in XLA_TEST_P()
259 auto input_literal = LiteralUtil::CreateFromArray(*a4x3); in XLA_TEST_P()
267 auto expected_literal = LiteralUtil::CreateFromArray(*expected); in XLA_TEST_P()
275 auto input_literal = LiteralUtil::CreateFromArray(Array2D<float>(0, 4)); in XLA_TEST_P()
289 auto input_literal = LiteralUtil::CreateFromArray(*a4x3); in XLA_TEST_P()
296 auto expected_literal = LiteralUtil::CreateFromArray(*expected); in XLA_TEST_P()
305 auto input_literal = LiteralUtil::CreateFromArray(Array2D<float>(6, 0)); in XLA_TEST_P()
312 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()
377 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()
1708 {LiteralUtil::CreateFromArray(param0), in XLA_TEST_P()
1709 LiteralUtil::CreateFromArray(param1)}, in XLA_TEST_P()
[all …]
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()
Dclient_library_test_base.h292 return AddParam(LiteralUtil::CreateFromArray(argument), builder); in AddParam()
307 return CreateConstantFromLiteral(LiteralUtil::CreateFromArray(array), in CreateConstantFromArray()
/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/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/mlir/xla/
Dmlir_hlo_to_hlo.cc799 return xla::LiteralUtil::CreateFromArray(source_data); \ in CreateLiteralFromAttr()
825 xla::LiteralUtil::CreateFromArray(source_data)); in CreateLiteralFromAttr()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h2034 LiteralUtil::CreateFromArray<NativeT>(values)); in ConstantFromArray()