Home
last modified time | relevance | path

Searched refs:_set_attr (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dcontrol_flow_util_v2.py121 op._set_attr("_lower_using_switch_merge", attr_value_pb2.AttrValue(b=True))
138 op._set_attr("_xla_propagate_compile_time_consts",
Dfunctional_ops.py640 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
Dcontrol_flow_ops.py2281 op._set_attr("frame_name",
/external/tensorflow/tensorflow/lite/python/
Dop_hint.py277 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(
/external/tensorflow/tensorflow/python/grappler/
Dmemory_optimizer_test.py76 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/python/tpu/
Dtpu.py373 op._set_attr(_TPU_REPLICATE_ATTR,
376 op._set_attr(
857 i.op._set_attr("_tpu_input_identity",
932 metadata._set_attr("host_compute_core", attr_value) # pylint: disable=protected-access
939 op._set_attr(_TPU_COMPILATION_STATUS_ATTR, attr_value) # pylint: disable=protected-access
1033 o.op._set_attr("_tpu_output_identity", attr_value_pb2.AttrValue(b=True))
1077 o.op._set_attr("_tpu_output_identity", attr_value_pb2.AttrValue(b=True))
/external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/
Dxla_sharding.py139 tensor.op._set_attr('_XlaSharding', attr_value)
/external/tensorflow/tensorflow/python/framework/
Dops.py2524 def _set_attr(self, attr_name, attr_value): member in Operation
2538 self._set_attr(attr_name, attr_value_pb2.AttrValue(func=func))
2546 self._set_attr(attr_name, attr_value_pb2.AttrValue(list=types_list))
2552 self._set_attr(attr_name, attr_value_pb2.AttrValue(list=shapes_list))
3531 op._set_attr(key, value) # pylint: disable=protected-access
3536 op._set_attr("_kernel", # pylint: disable=protected-access
3545 op._set_attr("_gradient_op_type", # pylint: disable=protected-access
3573 op._set_attr("_class", attr_value_pb2.AttrValue(
3590 op._set_attr("container", attr_value_pb2.AttrValue( # pylint: disable=protected-access
Dfunc_graph.py921 placeholder.op._set_attr( # pylint: disable=protected-access
931 placeholder.op._set_attr( # pylint: disable=protected-access
Dops_test.py471 op._set_attr("foo", attr_value_pb2.AttrValue(i=2))
/external/tensorflow/tensorflow/contrib/graph_editor/tests/
Dtransform_test.py47 c0.op._set_attr("_foo", attr_value_pb2.AttrValue(s=b"foo"))
/external/tensorflow/tensorflow/contrib/compiler/
Dxla.py192 op._set_attr(
/external/tensorflow/tensorflow/contrib/quantize/python/
Dfold_batch_norms.py273 bn_op._set_attr(
/external/tensorflow/tensorflow/python/kernel_tests/
Dfunctional_ops_test.py652 outputs[0].op._set_attr("_lower_using_switch_merge",