| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | reshape_test.cc | 238 auto input_literal = LiteralUtil::CreateFromArray(Array2D<float>(0, 2)); in XLA_TEST_P()
254 auto input_literal = LiteralUtil::CreateFromArray(*simple); in XLA_TEST_P()
263 auto expected_literal = LiteralUtil::CreateFromArray(*expected); in XLA_TEST_P()
272 auto input_literal = LiteralUtil::CreateFromArray(*a4x3); in XLA_TEST_P()
281 auto expected_literal = LiteralUtil::CreateFromArray(*expected); in XLA_TEST_P()
289 auto input_literal = LiteralUtil::CreateFromArray(Array2D<float>(0, 4)); in XLA_TEST_P()
304 auto input_literal = LiteralUtil::CreateFromArray(*a4x3); in XLA_TEST_P()
312 auto expected_literal = LiteralUtil::CreateFromArray(*expected); in XLA_TEST_P()
321 auto input_literal = LiteralUtil::CreateFromArray(Array2D<float>(6, 0)); in XLA_TEST_P()
329 LiteralUtil::CreateFromArray(Array4D<float>(2, 3, 0, 0)); in XLA_TEST_P()
[all …]
|
| H A D | reduce_window_test.cc | 171 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*res), {}, in XLA_TEST_P()
186 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*res), {}, in XLA_TEST_P()
200 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*res), {}, in XLA_TEST_P()
217 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*res), {}, in XLA_TEST_P()
239 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*result), {}, in XLA_TEST_P()
276 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*result), {}, in XLA_TEST_P()
301 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*result), {}, in XLA_TEST_P()
318 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*result), {}, in XLA_TEST_P()
356 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*expected), in XLA_TEST_P()
376 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*res), in XLA_TEST_P()
[all …]
|
| H A D | convolution_test.cc | 132 {LiteralUtil::CreateFromArray(input_data), in RunTest()
133 LiteralUtil::CreateFromArray(filter_data)}, in RunTest()
170 {LiteralUtil::CreateFromArray(input_data), in RunTest()
171 LiteralUtil::CreateFromArray(filter_data)}, in RunTest()
209 {LiteralUtil::CreateFromArray(input_data), in RunTest()
210 LiteralUtil::CreateFromArray(filter_data)}, in RunTest()
249 {LiteralUtil::CreateFromArray(input_data), in RunTest()
250 LiteralUtil::CreateFromArray(filter_data)}, in RunTest()
1516 {LiteralUtil::CreateFromArray(param0), in XLA_TEST_P()
1517 LiteralUtil::CreateFromArray(param1)}, in XLA_TEST_P()
[all …]
|
| H A D | client_library_test_base.h | 298 return AddParam(LiteralUtil::CreateFromArray(argument), builder); in AddParam()
313 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()
673 Literal literal = LiteralUtil::CreateFromArray(array); in CreateParameter()
|
| H A D | batch_normalization_test.cc | 263 {LiteralUtil::CreateFromArray(out), LiteralUtil::CreateR1<float>({4, 5}), in XLA_TEST_F()
317 {LiteralUtil::CreateFromArray(out), LiteralUtil::CreateR1<float>({4, 5}), in XLA_TEST_F()
447 {LiteralUtil::CreateFromArray(out), LiteralUtil::CreateR1<float>({0, 0}), in XLA_TEST_F()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | literal_util.h | 133 static Literal CreateFromArray(const Array<NativeT>& values);
405 /* static */ Literal LiteralUtil::CreateFromArray( in CreateFromArray() function
420 return CreateFromArray(values); in CreateR2FromArray2D()
432 return CreateFromArray(values); in CreateR3FromArray3D()
491 return CreateFromArray(values); in CreateR4FromArray4D()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | bfloat16_propagation_test.cc | 242 HloInstruction::CreateConstant(LiteralUtil::CreateFromArray(array_a))); in TEST_F()
244 HloInstruction::CreateConstant(LiteralUtil::CreateFromArray(array_b))); in TEST_F()
258 LiteralUtil::ConvertF32ToBF16(LiteralUtil::CreateFromArray(array_a)), in TEST_F()
261 LiteralUtil::ConvertF32ToBF16(LiteralUtil::CreateFromArray(array_b)), in TEST_F()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | sorting_test.cc | 126 CreateConstantFromLiteral(LiteralUtil::CreateFromArray(input), &builder); in XLA_TEST_F()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | xla_builder.h | 2754 LiteralUtil::CreateFromArray<NativeT>(values)); in ConstantFromArray()
|