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

  /external/tensorflow/tensorflow/compiler/xrt/kernels/
xrt_compile_ops.cc 116 std::vector<const xla::Shape*> argument_layout_ptrs(
120 argument_layout_ptrs[i] = &argument_layouts[i];
133 client->Compile(computation, argument_layout_ptrs, build_options);
  /external/tensorflow/tensorflow/compiler/xla/tools/
replay_computation.cc 99 std::vector<const Shape*> argument_layout_ptrs; local
103 argument_layout_ptrs.push_back(&argument_layouts.back());
107 return client->Compile(computation, argument_layout_ptrs, exec_build_options);

Completed in 93 milliseconds