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

  /external/tensorflow/tensorflow/compiler/tf2xla/
cpu_function_runtime.h 47 bool is_entry_parameter() const { return kind() == Kind::kEntryParameter; } function in class:tensorflow::cpu_function_runtime::BufferInfo
51 assert(is_entry_parameter());
81 return !is_entry_parameter() ||
cpu_function_runtime.cc 64 (buffer_infos[i].is_entry_parameter() && allocate_entry_params);
91 (buffer_infos[i].is_entry_parameter() && allocate_entry_params);
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
buffer_info_util.cc 46 if (buffer_infos[i].is_entry_parameter()) {
  /external/tensorflow/tensorflow/compiler/xla/tests/
local_client_aot_test_helper.cc 96 CHECK(result->buffer_infos()[0].is_entry_parameter()); // param buffer
  /external/tensorflow/tensorflow/compiler/xla/service/
buffer_assignment.cc 1006 const bool is_entry_parameter = local
    [all...]
  /external/tensorflow/tensorflow/compiler/aot/
codegen.cc 107 return buffer_info.is_entry_parameter();

Completed in 309 milliseconds