Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/tpu/
Dtpu.py538 op._set_attr(
649 op._set_attr("_xla_outside_compilation",
657 op._set_attr("_xla_outside_compilation",
883 real_input_shape.op._set_attr( # pylint: disable=protected-access
950 padded_input.op._set_attr( # pylint: disable=protected-access
1163 i.op._set_attr("_tpu_input_identity",
1238 metadata._set_attr("host_compute_core", attr_value) # pylint: disable=protected-access
1245 op._set_attr(_TPU_COMPILATION_STATUS_ATTR, attr_value) # pylint: disable=protected-access
1339 o.op._set_attr("_tpu_output_identity", attr_value_pb2.AttrValue(b=True))
1383 o.op._set_attr("_tpu_output_identity", attr_value_pb2.AttrValue(b=True))
/external/tensorflow/tensorflow/lite/python/
Dop_hint.py278 identity_op.op._set_attr(
282 identity_op.op._set_attr(
286 identity_op.op._set_attr(
288 identity_op.op._set_attr(OpHint.FUNCTION_LEVEL_ATTR,
291 identity_op.op._set_attr(
298 identity_op.op._set_attr(
302 identity_op.op._set_attr(
380 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/ops/
Dcontrol_flow_util_v2.py109 op._set_attr("_lower_using_switch_merge", attr_value_pb2.AttrValue(b=True))
126 op._set_attr("_xla_propagate_compile_time_consts",
Dfunctional_ops.py943 ret[0].op._set_attr("_input_hostmem", input_attr) # pylint: disable=protected-access
947 ret[0].op._set_attr("_output_hostmem", output_attr) # pylint: disable=protected-access
1073 ret[0].op._set_attr("_input_hostmem", input_attr) # pylint: disable=protected-access
1077 ret[0].op._set_attr("_output_hostmem", output_attr) # pylint: disable=protected-access
Dwhile_v2.py276 outputs[0].op._set_attr("_num_original_outputs",
375 while_op._set_attr("_num_original_outputs",
Dcontrol_flow_ops.py1526 op._set_attr("frame_name",
/external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/
Dxla_sharding.py145 tensor.op._set_attr('_XlaSharding', attr_value)
/external/tensorflow/tensorflow/python/framework/
Dops.py2274 def _set_attr(self, attr_name, attr_value): member in Operation
2293 self._set_attr(attr_name, attr_value_pb2.AttrValue(func=func))
2300 self._set_attr(attr_name, attr_value_pb2.AttrValue(list=funcs_list))
2309 self._set_attr(attr_name, attr_value_pb2.AttrValue(list=types_list))
2315 self._set_attr(attr_name, attr_value_pb2.AttrValue(list=shapes_list))
3387 op._set_attr(key, value) # pylint: disable=protected-access
3392 op._set_attr("_kernel", # pylint: disable=protected-access
3403 op._set_attr("_gradient_op_type", # pylint: disable=protected-access
3431 op._set_attr(
3450 op._set_attr("container", attr_value_pb2.AttrValue( # pylint: disable=protected-access
Dfunc_graph.py1220 placeholder.op._set_attr( # pylint: disable=protected-access
1243 placeholder.op._set_attr( # pylint: disable=protected-access
Dops_test.py774 op._set_attr("foo", attr_value_pb2.AttrValue(i=2))
/external/tensorflow/tensorflow/python/compiler/xla/
Dxla.py216 op._set_attr(
/external/tensorflow/tensorflow/python/kernel_tests/
Dfunctional_ops_test.py653 outputs[0].op._set_attr("_lower_using_switch_merge",
1208 outputs[0].op._set_attr("_lower_using_switch_merge",
1249 outputs[0].op._set_attr("_lower_using_switch_merge",
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dmap_test.py1057 dataset._variant_tensor.op._set_attr("use_inter_op_parallelism",