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

  /external/tensorflow/tensorflow/python/ops/
functional_ops.py 638 input_attr = attr_value_pb2.AttrValue()
639 input_attr.list.i.extend(hostmem)
640 ret[0].op._set_attr("_input_hostmem", input_attr) # pylint: disable=protected-access
768 input_attr = attr_value_pb2.AttrValue()
769 input_attr.list.i.extend([num_for_params + i for i in hostmem])
770 ret[0].op._set_attr("_input_hostmem", input_attr) # pylint: disable=protected-access
    [all...]
  /external/tensorflow/tensorflow/core/framework/
op_kernel.cc 551 const auto input_attr = params_->input_alloc_attrs == nullptr local
554 if (!output_attr.IsEqualOrLessRestrictiveThan(input_attr)) {
    [all...]

Completed in 331 milliseconds