Home
last modified time | relevance | path

Searched refs:get_attr (Results 1 – 25 of 79) sorted by relevance

1234

/external/tensorflow/tensorflow/python/ops/
Dnn_grad.py49 dilations=op.get_attr("dilations"),
50 strides=op.get_attr("strides"),
51 padding=op.get_attr("padding"),
52 explicit_paddings=op.get_attr("explicit_paddings"),
53 use_cudnn_on_gpu=op.get_attr("use_cudnn_on_gpu"),
54 data_format=op.get_attr("data_format").decode()),
58 dilations=op.get_attr("dilations"),
59 strides=op.get_attr("strides"),
60 padding=op.get_attr("padding"),
61 explicit_paddings=op.get_attr("explicit_paddings"),
[all …]
Dcudnn_rnn_grad.py27 if not op.get_attr("is_training"):
42 dropout=op.get_attr("dropout"),
43 seed=op.get_attr("seed"),
44 seed2=op.get_attr("seed2"),
45 rnn_mode=op.get_attr("rnn_mode"),
46 input_mode=op.get_attr("input_mode"),
47 direction=op.get_attr("direction"))
52 if not op.get_attr("is_training"):
68 dropout=op.get_attr("dropout"),
69 seed=op.get_attr("seed"),
[all …]
Darray_grad.py44 return array_ops.unstack(grad, num=op.get_attr("N"), axis=op.get_attr("axis"))
50 return array_ops.stack(grads, axis=op.get_attr("axis"))
291 begin_mask=op.get_attr("begin_mask"),
292 end_mask=op.get_attr("end_mask"),
293 ellipsis_mask=op.get_attr("ellipsis_mask"),
294 new_axis_mask=op.get_attr("new_axis_mask"),
295 shrink_axis_mask=op.get_attr("shrink_axis_mask")), None, None, None
310 begin_mask=op.get_attr("begin_mask"),
311 end_mask=op.get_attr("end_mask"),
312 ellipsis_mask=op.get_attr("ellipsis_mask"),
[all …]
Dimage_grad.py48 align_corners=op.get_attr("align_corners"),
49 half_pixel_centers=op.get_attr("half_pixel_centers"))
67 align_corners=op.get_attr("align_corners"),
68 half_pixel_centers=op.get_attr("half_pixel_centers"))
89 kernel_type=op.get_attr("kernel_type"),
90 antialias=op.get_attr("antialias"))
111 align_corners=op.get_attr("align_corners"),
112 half_pixel_centers=op.get_attr("half_pixel_centers"))
141 grad, op.inputs[1], op.inputs[2], image_shape, T=op.get_attr("T"),
142 method=op.get_attr("method"))
Dtensor_array_grad.py106 dtype = op.get_attr("dtype")
132 dtype = op.get_attr("T")
169 dtype = op.get_attr("dtype")
193 dtype = op.get_attr("T")
230 dtype = op.get_attr("dtype")
254 dtype = op.get_attr("T")
Dnccl_ops.py64 if op.get_attr('reduction') != b'sum':
68 num_devices = op.get_attr('num_devices')
69 shared_name = op.get_attr('shared_name') + b'_grad'
163 if op.get_attr('reduction') != b'sum':
Dimage_ops.py247 interpolation = op.get_attr("interpolation")
248 fill_mode = op.get_attr("fill_mode")
281 interpolation = op.get_attr("interpolation")
282 fill_mode = op.get_attr("fill_mode")
Dcritical_section_ops.py72 return op.get_attr("_class")
383 and x.op.get_attr("shared_name")
384 and (x.op.get_attr("shared_name") ==
385 self._handle.op.get_attr("shared_name"))
Dlinalg_grad.py296 equation = op.get_attr("equation")
555 adjoint_a = op.get_attr("adjoint")
626 fast = op.get_attr("fast")
650 adjoint_a = op.get_attr("adjoint")
651 lower_a = op.get_attr("lower")
682 adjoint_a = op.get_attr("adjoint")
683 lower_a = op.get_attr("lower")
727 compute_v = op.get_attr("compute_v")
772 compute_v = op.get_attr("compute_v")
827 if not op.get_attr("compute_uv"):
[all …]
Dmath_grad.py58 if not op.get_attr("keep_dims"):
210 if not op.get_attr("keep_dims"):
224 if not op.get_attr("keep_dims"):
285 if not op.get_attr("keep_dims"):
1685 t_a = op.get_attr("transpose_a")
1686 t_b = op.get_attr("transpose_b")
1701 t_a = op.get_attr("transpose_a")
1702 t_b = op.get_attr("transpose_b")
1729 t_a = op.get_attr("transpose_a")
1730 t_b = op.get_attr("transpose_b")
[all …]
/external/tensorflow/tensorflow/python/compiler/xla/
Djit_test.py81 self.assertFalse(v_false_1_t.op.get_attr("_XlaCompile"))
87 self.assertFalse(v_true_1_t_sampler_op.get_attr("_XlaCompile"))
88 self.assertTrue(v_all_true_t_sampler_op.get_attr("_XlaCompile"))
90 self.assertTrue(v_true_1_t.op.get_attr("_XlaCompile"))
91 self.assertTrue(v_all_true_t.op.get_attr("_XlaCompile"))
121 self.assertEqual(b"jit_scope_0", a1.op.get_attr("_XlaScope"))
122 self.assertEqual(b"jit_scope_1", a2.op.get_attr("_XlaScope"))
123 self.assertEqual(b"jit_scope_1", a3.op.get_attr("_XlaScope"))
124 self.assertEqual(b"jit_scope_1", a4.op.get_attr("_XlaScope"))
125 self.assertEqual(b"jit_scope_1", a5.op.get_attr("_XlaScope"))
[all …]
/external/tensorflow/tensorflow/python/ops/linalg/sparse/
Dsparse_csr_matrix_grad.py32 grad, type=op.get_attr("T")))
74 grad, type=op.get_attr("type"), conjugate=op.get_attr("conjugate"))
82 softmax, grad_softmax, type=op.get_attr("type"))
95 t_a = op.get_attr("transpose_a")
96 t_b = op.get_attr("transpose_b")
97 adj_a = op.get_attr("adjoint_a")
98 adj_b = op.get_attr("adjoint_b")
99 transpose_output = op.get_attr("transpose_output")
100 conjugate_output = op.get_attr("conjugate_output")
175 t_a = op.get_attr("transpose_a")
[all …]
/external/tensorflow/tensorflow/python/util/
Dexample_parser_configuration.py50 num_sparse = parse_example_op.get_attr("Nsparse")
51 num_dense = parse_example_op.get_attr("Ndense")
54 sparse_types = parse_example_op.get_attr("sparse_types")
55 dense_types = parse_example_op.get_attr("Tdense")
56 dense_shapes = parse_example_op.get_attr("dense_shapes")
139 dense_types = parse_example_op.get_attr("Tdense")
140 num_sparse = parse_example_op.get_attr("num_sparse")
141 sparse_types = parse_example_op.get_attr("sparse_types")
142 ragged_value_types = parse_example_op.get_attr("ragged_value_types")
143 ragged_split_types = parse_example_op.get_attr("ragged_split_types")
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/
Dops_defs.py78 act = op.get_attr('act')
96 dilations = [1, op.get_attr('dilation_w'), op.get_attr('dilation_h'), 1]
97 strides = [1, op.get_attr('stride_w'), op.get_attr('stride_h'), 1]
98 padding = op.get_attr('padding')
142 act = op.get_attr('act')
186 filter_width = op.get_attr('filter_width')
187 filter_height = op.get_attr('filter_height')
188 stride_w = op.get_attr('stride_w')
189 stride_h = op.get_attr('stride_h')
190 padding = op.get_attr('padding')
/external/tensorflow/tensorflow/python/eager/
Dfunction_argument_naming_test.py56 [inp.op.get_attr('_user_specified_name') for inp in fn_op.inputs])
80 [inp.op.get_attr('_user_specified_name') for inp in fn_op.inputs])
99 [inp.op.get_attr('_user_specified_name') for inp in fn_op.inputs])
115 [inp.op.get_attr('_user_specified_name') for inp in fn_op2.inputs])
128 [inp.op.get_attr('_user_specified_name') for inp in fn_op3.inputs])
147 [inp.op.get_attr('_user_specified_name') for inp in class_op.inputs])
157 [inp.op.get_attr('_user_specified_name') for inp in method_op.inputs])
169 [inp.op.get_attr('_user_specified_name') for inp in method_op.inputs])
191 [inp.op.get_attr('_user_specified_name') for inp in method_op.inputs])
198 [inp.op.get_attr('_user_specified_name') for inp in method_op2.inputs])
[all …]
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py898 def get_attr(self, attr): member in _PforInput
899 return self._op.get_attr(attr)
1028 flat_attrs.append(op.get_attr(str(a)))
1889 is_training = pfor_input.get_attr("is_training")
1915 data_format = pfor_input.get_attr("data_format")
1942 data_format = pfor_input.get_attr("data_format")
1990 strides = pfor_input.get_attr("strides")
1991 padding = pfor_input.get_attr("padding")
1992 use_cudnn_on_gpu = pfor_input.get_attr("use_cudnn_on_gpu")
1993 data_format = pfor_input.get_attr("data_format")
[all …]
/external/tensorflow/tensorflow/python/tpu/ops/
Dtpu_ops.py87 concat_dimension=op.get_attr("split_dimension"),
88 split_dimension=op.get_attr("concat_dimension"),
89 split_count=op.get_attr("split_count")), None
169 table_id = activations_op.get_attr("table_id")
170 lookup_id = activations_op.get_attr("lookup_id")
/external/tensorflow/tensorflow/python/framework/
Dauto_control_deps_utils.py69 read_only_input_indices = op.get_attr(READ_ONLY_RESOURCE_INPUTS_ATTR)
109 read_only_input_indices = op.get_attr(READ_ONLY_RESOURCE_INPUTS_ATTR)
148 read_only_input_indices = op.get_attr(READ_ONLY_RESOURCE_INPUTS_ATTR)
Dtensor_util.py684 return MakeNdarray(tensor.op.get_attr("value"))
723 cast_dtype = dtypes.as_dtype(tensor.op.get_attr("DstT"))
755 if tensor.op.get_attr("axis") != 0:
765 if tensor.op.get_attr("axis") != 0:
776 split = np.split(value, tensor.op.get_attr("num_split"), dim)
930 cast_dtype = dtypes.as_dtype(tensor.op.get_attr("DstT"))
945 assert tensor.op.get_attr("axis") == 0
985 begin_mask = tensor.op.get_attr("begin_mask")
988 end_mask = tensor.op.get_attr("end_mask")
992 ellipsis_mask = tensor.op.get_attr("ellipsis_mask")
[all …]
Dauto_control_deps.py156 return [op.get_attr("_collective_manager_id")]
161 return op.get_attr(utils.COLLECTIVE_MANAGER_IDS)
390 collective_manager_scopes_opened[op.get_attr(
/external/tensorflow/tensorflow/python/tpu/
Dtpu_test.py154 a = graph.get_operation_by_name("import/a").get_attr(
157 x = graph.get_operation_by_name("import/x").get_attr(
164 graph.get_operation_by_name("import/b").get_attr(
169 graph.get_operation_by_name("import/y").get_attr(
/external/llvm-project/lldb/source/Plugins/SymbolFile/DWARF/
DDWARFAttribute.h29 dw_attr_t get_attr() const { return m_attr; } in get_attr() function
62 return m_infos[i].attr.get_attr(); in AttributeAtIndex()
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_conversion_ops.py67 row_partition_types = op.get_attr("row_partition_types")
158 batched_input = (op.get_attr("output_ragged_rank") > 0)
/external/libnetfilter_conntrack/include/internal/
Dextern.h5 extern const get_attr get_attr_array[];
Dtypes.h11 typedef const void *(*get_attr)(const struct nf_conntrack *ct); typedef

1234