HomeSort by relevance Sort by last modified time
    Searched refs:CreateFromArray (Results 1 - 7 of 7) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/tests/
reshape_test.cc 227 auto input_literal = LiteralUtil::CreateFromArray(Array2D<float>(0, 2));
242 auto input_literal = LiteralUtil::CreateFromArray(*simple);
250 auto expected_literal = LiteralUtil::CreateFromArray(*expected);
259 auto input_literal = LiteralUtil::CreateFromArray(*a4x3);
267 auto expected_literal = LiteralUtil::CreateFromArray(*expected);
275 auto input_literal = LiteralUtil::CreateFromArray(Array2D<float>(0, 4));
289 auto input_literal = LiteralUtil::CreateFromArray(*a4x3);
296 auto expected_literal = LiteralUtil::CreateFromArray(*expected);
305 auto input_literal = LiteralUtil::CreateFromArray(Array2D<float>(6, 0));
312 LiteralUtil::CreateFromArray(Array4D<float>(2, 3, 0, 0))
    [all...]
reduce_window_test.cc 164 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*res), {},
179 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*res), {},
193 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*res), {},
210 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*res), {},
232 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*result), {},
255 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*result), {},
280 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*result), {},
297 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*result), {},
335 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*expected),
354 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*res)
    [all...]
convolution_test.cc 133 {LiteralUtil::CreateFromArray(input_data),
134 LiteralUtil::CreateFromArray(filter_data)},
167 {LiteralUtil::CreateFromArray(input_data),
168 LiteralUtil::CreateFromArray(filter_data)},
202 {LiteralUtil::CreateFromArray(input_data),
203 LiteralUtil::CreateFromArray(filter_data)},
234 {LiteralUtil::CreateFromArray(input_data),
235 LiteralUtil::CreateFromArray(filter_data)},
    [all...]
client_library_test_base.h 292 return AddParam(LiteralUtil::CreateFromArray(argument), builder);
307 return CreateConstantFromLiteral(LiteralUtil::CreateFromArray(array),
  /external/tensorflow/tensorflow/compiler/xla/
literal_util.h 164 static Literal CreateFromArray(const Array<NativeT>& values);
446 /* static */ Literal LiteralUtil::CreateFromArray(
461 return CreateFromArray(values);
473 return CreateFromArray(values);
532 return CreateFromArray(values);
  /external/tensorflow/tensorflow/compiler/xla/service/
bfloat16_propagation_test.cc 240 HloInstruction::CreateConstant(LiteralUtil::CreateFromArray(array_a)));
242 HloInstruction::CreateConstant(LiteralUtil::CreateFromArray(array_b)));
256 LiteralUtil::ConvertF32ToBF16(LiteralUtil::CreateFromArray(array_a)),
259 LiteralUtil::ConvertF32ToBF16(LiteralUtil::CreateFromArray(array_b)),
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/client/
xla_builder.h     [all...]

Completed in 1852 milliseconds