Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/
Dcontrol_flow_util_v2.py120 op._set_attr("_lower_using_switch_merge",
128 op._set_attr("_lower_using_switch_merge", attr_value_pb2.AttrValue(b=True))
145 op._set_attr("_xla_propagate_compile_time_consts",
Dfunctional_ops.py985 ret[0].op._set_attr("_input_hostmem", input_attr) # pylint: disable=protected-access
989 ret[0].op._set_attr("_output_hostmem", output_attr) # pylint: disable=protected-access
1115 ret[0].op._set_attr("_input_hostmem", input_attr) # pylint: disable=protected-access
1119 ret[0].op._set_attr("_output_hostmem", output_attr) # pylint: disable=protected-access
Dwhile_v2.py388 while_op._set_attr("_num_original_outputs",
451 while_op._set_attr("_num_original_outputs",
Dcontrol_flow_ops.py1572 op._set_attr("frame_name",
/external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/
Dxla_sharding.py198 tensor.op._set_attr('_XlaSharding',
210 operation._set_attr('_XlaSharding', attr_value)
253 to_tensor.op._set_attr('_XlaSharding', attr_value)
/external/tensorflow/tensorflow/python/grappler/
Dmemory_optimizer_test.py77 d.op._set_attr('_swap_to_host', attr_value_pb2.AttrValue(i=0))
275 current_activation.op._set_attr(
281 current_activation.op._set_attr(
284 current_activation.op._set_attr(
/external/tensorflow/tensorflow/lite/python/
Dop_hint.py289 identity_op.op._set_attr(
293 identity_op.op._set_attr(
297 identity_op.op._set_attr(
299 identity_op.op._set_attr(OpHint.FUNCTION_LEVEL_ATTR,
302 identity_op.op._set_attr(
309 identity_op.op._set_attr(
313 identity_op.op._set_attr(
391 dest_op.op._set_attr(name, _attr_value_pb2.AttrValue(
/external/tensorflow/tensorflow/python/tpu/
Dtpu.py593 op._set_attr(
713 op._set_attr("_xla_outside_compilation",
721 op._set_attr("_xla_outside_compilation",
1044 real_input_shape.op._set_attr( # pylint: disable=protected-access
1114 padded_input.op._set_attr( # pylint: disable=protected-access
1393 pivot._set_attr(_PIVOT_FOR_CLUSTER, # pylint: disable=protected-access
1432 i.op._set_attr("_tpu_input_identity",
1518 metadata._set_attr("host_compute_core", attr_value) # pylint: disable=protected-access
1525 op._set_attr(_TPU_COMPILATION_STATUS_ATTR, attr_value) # pylint: disable=protected-access
1649 o.op._set_attr("_tpu_output_identity", attr_value_pb2.AttrValue(b=True))
[all …]
Dtpu_embedding_v2.py92 …op._set_attr(_NAME_KEY, attr_value_pb2.AttrValue(s=compat.as_bytes(name))) # pylint: disable=prot…
/external/tensorflow/tensorflow/python/framework/
Dops.py2484 def _set_attr(self, attr_name, attr_value): member in Operation
2503 self._set_attr(attr_name, attr_value_pb2.AttrValue(func=func))
2510 self._set_attr(attr_name, attr_value_pb2.AttrValue(list=funcs_list))
2519 self._set_attr(attr_name, attr_value_pb2.AttrValue(list=types_list))
2525 self._set_attr(attr_name, attr_value_pb2.AttrValue(list=shapes_list))
3605 op._set_attr(key, value) # pylint: disable=protected-access
3610 op._set_attr("_kernel", # pylint: disable=protected-access
3621 op._set_attr("_gradient_op_type", # pylint: disable=protected-access
3654 op._set_attr(
3673 op._set_attr("container", attr_value_pb2.AttrValue( # pylint: disable=protected-access
[all …]
Dfunc_graph.py660 placeholder.op._set_attr( # pylint: disable=protected-access
1235 placeholder.op._set_attr( # pylint: disable=protected-access
1258 placeholder.op._set_attr( # pylint: disable=protected-access
Dops_test.py948 op._set_attr("foo", attr_value_pb2.AttrValue(i=2))
/external/tensorflow/tensorflow/python/compiler/xla/
Dxla.py221 op._set_attr(
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py450 grad_sharding.op._set_attr("_XlaSharding",
/external/tensorflow/tensorflow/python/kernel_tests/
Dfunctional_ops_test.py678 outputs[0].op._set_attr("_lower_using_switch_merge",
1232 outputs[0].op._set_attr("_lower_using_switch_merge",
1273 outputs[0].op._set_attr("_lower_using_switch_merge",
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dmap_test.py1066 dataset._variant_tensor.op._set_attr("use_inter_op_parallelism",