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

  /external/tensorflow/tensorflow/compiler/xla/tests/
compilation_cache_test.cc 144 auto rowmaj_array = LiteralUtil::CreateR2WithLayout(
149 auto colmaj_array = LiteralUtil::CreateR2WithLayout(
client_test.cc 59 Literal expected_literal = LiteralUtil::CreateR2WithLayout<int32>(
params_test.cc 459 Literal literal = LiteralUtil::CreateR2WithLayout<float>(
471 Literal literal = LiteralUtil::CreateR2WithLayout<float>(
compute_constant_test.cc 259 Literal expected_literal = LiteralUtil::CreateR2WithLayout<int32>(
transfer_manager_test.cc 155 Literal literal = LiteralUtil::CreateR2WithLayout<float>(
map_test.cc 368 Literal param0_literal = LiteralUtil::CreateR2WithLayout(
373 Literal param1_literal = LiteralUtil::CreateR2WithLayout(
local_client_execute_test.cc 131 auto x_array = LiteralToShapedBuffer(LiteralUtil::CreateR2WithLayout(
137 auto y_array = LiteralToShapedBuffer(LiteralUtil::CreateR2WithLayout(
    [all...]
dot_operation_test.cc 505 ->TransferToServer(LiteralUtil::CreateR2WithLayout<complex64>(
510 ->TransferToServer(LiteralUtil::CreateR2WithLayout<complex64>(
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/
literal_util.h 80 static Literal CreateR2WithLayout(
331 /* static */ Literal LiteralUtil::CreateR2WithLayout(
346 return CreateR2WithLayout(values, LayoutUtil::GetDefaultLayoutForR2());
literal_test.cc 745 auto mat_dim0minor = LiteralUtil::CreateR2WithLayout<int32>(
756 auto mat_dim0major = LiteralUtil::CreateR2WithLayout<int32>(
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_cse_test.cc 84 HloInstruction::CreateConstant(LiteralUtil::CreateR2WithLayout<float>(
87 HloInstruction::CreateConstant(LiteralUtil::CreateR2WithLayout<float>(
116 HloInstruction::CreateConstant(LiteralUtil::CreateR2WithLayout<float>(
119 HloInstruction::CreateConstant(LiteralUtil::CreateR2WithLayout<float>(
layout_assignment_test.cc 165 auto constant_literal1 = LiteralUtil::CreateR2WithLayout<float>(
167 auto constant_literal2 = LiteralUtil::CreateR2WithLayout<float>(
216 HloInstruction::CreateConstant(LiteralUtil::CreateR2WithLayout<float>(
219 HloInstruction::CreateConstant(LiteralUtil::CreateR2WithLayout<float>(
253 HloInstruction::CreateConstant(LiteralUtil::CreateR2WithLayout<float>(
256 HloInstruction::CreateConstant(LiteralUtil::CreateR2WithLayout<float>(
    [all...]
hlo_evaluator_test.cc 541 auto input_literal = LiteralUtil::CreateR2WithLayout<int32>(
543 auto expected = LiteralUtil::CreateR2WithLayout<float>(
    [all...]
  /external/tensorflow/tensorflow/compiler/xrt/tests/
raw_api_test.cc 105 auto array = xla::LiteralUtil::CreateR2WithLayout<float>(v, layout);
    [all...]

Completed in 1403 milliseconds