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

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_executable.cc 255 auto param_no = allocation.parameter_number(); local
257 arguments[param_no]->buffer(allocation.param_shape_index());
266 allocation.param_shape_index().ToString(), param_no);
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_computation.cc 79 int64 param_no = instruction->parameter_number(); local
80 CHECK(param_no >= 0 && param_no < parameter_count)
82 << parameter_count << "), got " << param_no; local
83 CHECK(param_instructions_[param_no] == nullptr)
84 << "\nERROR: parameter number " << param_no
86 param_instructions_[param_no] = instruction.get();
145 Status HloComputation::RemoveParameter(int64 param_no) {
146 CHECK_GE(param_no, 0);
147 CHECK_LT(param_no, param_instructions_.size())
182 const int64 param_no = i - removed; local
    [all...]

Completed in 642 milliseconds