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

  /external/tensorflow/tensorflow/python/ops/
control_flow_util_v2.py 121 op._set_attr("_lower_using_switch_merge", attr_value_pb2.AttrValue(b=True))
138 op._set_attr("_xla_propagate_compile_time_consts",
functional_ops.py 640 ret[0].op._set_attr("_input_hostmem", input_attr) # pylint: disable=protected-access
644 ret[0].op._set_attr("_output_hostmem", output_attr) # pylint: disable=protected-access
770 ret[0].op._set_attr("_input_hostmem", input_attr) # pylint: disable=protected-access
774 ret[0].op._set_attr("_output_hostmem", output_attr) # pylint: disable=protected-access
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/
xla_sharding.py 139 tensor.op._set_attr('_XlaSharding', attr_value)
  /external/tensorflow/tensorflow/python/grappler/
memory_optimizer_test.py 76 d.op._set_attr('_swap_to_host', attr_value_pb2.AttrValue(i=0))
274 current_activation.op._set_attr(
280 current_activation.op._set_attr(
283 current_activation.op._set_attr(
  /external/tensorflow/tensorflow/lite/python/
op_hint.py 277 identity_op.op._set_attr(
281 identity_op.op._set_attr(
285 identity_op.op._set_attr(
287 identity_op.op._set_attr(OpHint.FUNCTION_LEVEL_ATTR,
290 identity_op.op._set_attr(
297 identity_op.op._set_attr(
301 identity_op.op._set_attr(
379 dest_op.op._set_attr(name, _attr_value_pb2.AttrValue(
    [all...]
  /external/tensorflow/tensorflow/python/tpu/
tpu.py 373 op._set_attr(_TPU_REPLICATE_ATTR,
376 op._set_attr(
    [all...]
  /external/tensorflow/tensorflow/python/framework/
ops.py 2524 def _set_attr(self, attr_name, attr_value): member in class:Operation
    [all...]
func_graph.py     [all...]
ops_test.py 471 op._set_attr("foo", attr_value_pb2.AttrValue(i=2))
    [all...]
  /external/tensorflow/tensorflow/contrib/compiler/
xla.py 192 op._set_attr(
  /external/tensorflow/tensorflow/contrib/graph_editor/tests/
transform_test.py 47 c0.op._set_attr("_foo", attr_value_pb2.AttrValue(s=b"foo"))
  /external/tensorflow/tensorflow/contrib/quantize/python/
fold_batch_norms.py 273 bn_op._set_attr(
    [all...]
  /external/tensorflow/tensorflow/python/kernel_tests/
functional_ops_test.py 652 outputs[0].op._set_attr("_lower_using_switch_merge",
    [all...]

Completed in 448 milliseconds