| /external/tensorflow/tensorflow/python/framework/ |
| D | tensor_spec.py | 56 self._dtype = dtypes.as_dtype(dtype)
67 return self._dtype
76 self._dtype.is_compatible_with(spec_or_value.dtype) and
90 and self._dtype == other._dtype
97 if (type(self) is not type(other)) or (self._dtype != other.dtype):
101 return type(self)(shape, self._dtype, name)
104 return (self._shape, self._dtype, self._name)
107 return self._dtype
183 value = ops.convert_to_tensor(value, self._dtype)
186 "and shape %s." % (value, self._dtype, self._shape))
[all …]
|
| D | sparse_tensor.py | 312 self._dtype = dtypes.as_dtype(dtype)
315 return (self._shape, self._dtype)
320 return self._dtype
333 tensor_spec.TensorSpec([num_values], self._dtype),
375 tensor_list = gen_sparse_ops.deserialize_sparse(tensor_list[0], self._dtype)
402 self._dtype)
407 return SparseTensorSpec(self._shape[1:], self._dtype)
410 return self._dtype
|
| D | ops.py | 378 self._dtype = dtypes.as_dtype(dtype)
413 return self._dtype
830 (", dtype=%s" % self._dtype.name) if self._dtype else "",
835 self._dtype.name)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | tensor_array_ops.py | 125 self._dtype = dtypes.as_dtype(dtype).base_dtype
176 return self._dtype
243 dtype=self._dtype,
259 dtype=self._dtype,
270 value, preferred_dtype=self._dtype, name="value")
271 _check_dtypes(value, self._dtype)
303 dtype=self._dtype,
315 dtype=self._dtype,
337 value, preferred_dtype=self._dtype, name="value")
338 _check_dtypes(value, self._dtype)
[all …]
|
| D | variable_scope.py | 1157 self._dtype = dtype
1185 return self._dtype
1221 self._dtype = dtype
1319 dtype = self._dtype
1366 dtype = self._dtype
1918 self._dtype = dtype
1955 if self._dtype is not None:
1956 variable_scope_object.set_dtype(self._dtype)
2014 if self._dtype is not None:
2015 variable_scope_object.set_dtype(self._dtype)
[all …]
|
| D | resource_variable_ops.py | 423 self._dtype = dtypes.as_dtype(dtype)
503 dtype=self._dtype,
514 return self._dtype
666 self._handle, self._dtype)
667 _maybe_set_handle_data(self._dtype, self._handle, result)
706 self._handle, indices, dtype=self._dtype, name=name)
708 if self._dtype == dtypes.variant:
725 self._handle, indices, dtype=self._dtype, name=name)
1890 self._dtype = dtypes.as_dtype(self._handle.op.get_attr("dtype"))
2052 self._handle, self._dtype)
[all …]
|
| D | session_ops.py | 58 self._dtype = dtype
98 self._dtype)
|
| D | data_flow_ops.py | 1259 self._dtype = dtype
1283 return self._dtype
1375 grad = ops.convert_to_tensor(grad, self._dtype)
1405 self._accumulator_ref, num_required, dtype=self._dtype, name=name)
1541 self._accumulator_ref, num_required, dtype=self._dtype, name=name)
1565 self._accumulator_ref, num_required, dtype=self._dtype, name=name)
|
| D | map_fn.py | 540 return ragged_tensor.RaggedTensorSpec(None, spec._dtype, spec._ragged_rank,
|
| D | variables.py | 2950 self._dtype = dtype
3032 return self._dtype
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | metrics.py | 163 self._dtype = K.floatx() if dtype is None else dtypes.as_dtype(dtype).name
243 return self._dtype
309 dtype=self._dtype if dtype is None else dtype,
380 values = math_ops.cast(values, self._dtype)
382 sample_weight = math_ops.cast(sample_weight, self._dtype)
412 num_values = math_ops.cast(array_ops.size(values), self._dtype)
415 num_values = math_ops.cast(array_ops.size(values), self._dtype)
557 normalizer = math_ops.cast(normalizer, self._dtype)
573 y_true = math_ops.cast(y_true, self._dtype)
574 y_pred = math_ops.cast(y_pred, self._dtype)
[all …]
|
| D | metrics_test.py | 632 self.assertEqual(cosine_obj._dtype, dtypes.int32)
637 self.assertEqual(cosine_obj2._dtype, dtypes.int32)
675 self.assertEqual(mae_obj._dtype, dtypes.int32)
680 self.assertEqual(mae_obj2._dtype, dtypes.int32)
714 self.assertEqual(mape_obj._dtype, dtypes.int32)
720 self.assertEqual(mape_obj2._dtype, dtypes.int32)
753 self.assertEqual(mse_obj._dtype, dtypes.int32)
758 self.assertEqual(mse_obj2._dtype, dtypes.int32)
792 self.assertEqual(msle_obj._dtype, dtypes.int32)
798 self.assertEqual(msle_obj2._dtype, dtypes.int32)
[all …]
|
| /external/tensorflow/tensorflow/python/ops/linalg/sparse/ |
| D | sparse_csr_matrix_ops.py | 334 self._dtype = value.dtype
341 self._dtype = value.dtype
363 ret._dtype = self._dtype
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor.py | 2203 return self._dtype
2309 self._dtype = dtypes.as_dtype(dtype)
2332 self._shape, self._dtype).is_compatible_with(spec_or_value)
2339 return (self._shape, self._dtype, self._ragged_rank,
2342 return (self._shape, self._dtype, self._ragged_rank,
2351 return [tensor_spec.TensorSpec(self._shape, self._dtype)]
2357 flat_values_spec = tensor_spec.TensorSpec(flat_values_shape, self._dtype)
2441 dtype=self._dtype,
2459 self._dtype, self._ragged_rank + 1, self._row_splits_dtype)
2468 return RaggedTensorSpec(self._shape[1:], self._dtype, self._ragged_rank - 1,
[all …]
|
| D | row_partition.py | 1064 self._dtype = dtypes.as_dtype(dtype)
1065 if self._dtype not in (dtypes.int32, dtypes.int64):
1105 return (self._nrows, self._nvals, self._uniform_row_length, self._dtype)
1129 return self._dtype
1135 return tensor_spec.TensorSpec(row_splits_shape, self._dtype)
|
| D | ragged_tensor_supported_values_test.py | 346 self.assertEqual(spec1._dtype, dtypes.float32)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | sharded_variable.py | 294 self._dtype = first_var.dtype
323 shape=self._shape, dtype=self._dtype, name=self._name)
371 dtype=self._dtype,
515 return self._dtype
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | keras_tensor.py | 351 return self._type_spec._dtype # pylint: disable=protected-access
505 self._dtype = dtype
|
| D | base_layer_v1.py | 1828 def _dtype(self): member in Layer
1834 @_dtype.setter
1835 def _dtype(self, value): member in Layer
|
| D | base_layer.py | 2462 def _dtype(self): member in Layer
2468 @_dtype.setter
2469 def _dtype(self, value): member in Layer
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | autocast_variable.py | 556 self._dtype = dtype
560 _autocast_dtype.dtype = self._dtype
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | bijector_impl.py | 572 self._dtype = dtype
641 return self._dtype
|
| D | distribution.py | 471 self._dtype = dtype
569 return self._dtype
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator.py | 233 self._dtype = dtypes.as_dtype(dtype).base_dtype if dtype else dtype
258 return self._dtype
|
| /external/tensorflow/tensorflow/python/ops/structured/ |
| D | structured_tensor.py | 916 dtype=typespec._dtype,
|