| /external/tensorflow/tensorflow/python/data/util/ |
| structure.py | 431 self._dtype = dtypes.as_dtype(dtype)
440 return [self._dtype]
444 self._dtype.is_compatible_with(other._dtype) and
450 "and shape %s." % (value, self._dtype, self._shape))
463 "%s." % (flat_value[0], self._dtype, self._shape))
481 return self._dtype
491 self._dtype,
497 return TensorStructure(self._dtype, self._shape[1:])
505 self._dtype = dtypes.as_dtype(dtype
[all...] |
| /external/tensorflow/tensorflow/python/framework/ |
| tensor_spec.py | 38 __slots__ = ["_shape", "_shape_tuple", "_dtype", "_name"]
57 self._dtype = dtypes.as_dtype(dtype)
81 return self._dtype
100 return (self._dtype.is_compatible_with(spec_or_tensor.dtype) and
119 return TensorSpec, (self._shape, self._dtype, self._name)
217 return BoundedTensorSpec, (self._shape, self._dtype, self._minimum,
|
| ops.py | 326 self._dtype = dtypes.as_dtype(dtype)
344 return self._dtype
603 (", dtype=%s" % self._dtype.name)
604 if self._dtype else "", (", device=%s" % self.device)
609 self._dtype.name)
[all...] |
| /external/tensorflow/tensorflow/contrib/checkpoint/python/ |
| split_dependency.py | 54 self._dtype = dtype
88 dtype=self._dtype,
|
| /external/tensorflow/tensorflow/python/ops/ |
| tensor_array_ops.py | 120 self._dtype = dtype
174 return self._dtype
225 dtype=self._dtype,
249 dtype=self._dtype,
263 dtype=self._dtype,
284 dtype=self._dtype,
309 dtype=self._dtype,
326 dtype=self._dtype,
357 dtype=self._dtype,
388 dtype=self._dtype,
[all...] |
| matmul_benchmark_test.py | 115 for _dtype in dtypes:
121 BuildGraphTest(_n, _m, _k, _transpose_a, _transpose_b, _dtype))
123 RunGraphTest(_n, _m, _k, _transpose_a, _transpose_b, _dtype))
|
| resource_variable_ops.py | 565 self._dtype = initial_value.dtype.base_dtype
685 self._dtype = dtypes.as_dtype(self._handle.op.get_attr("dtype"))
720 dtype=self._dtype,
729 return self._dtype
[all...] |
| session_ops.py | 58 self._dtype = dtype
99 self._dtype)
|
| variable_scope.py | [all...] |
| data_flow_ops.py | [all...] |
| /external/tensorflow/tensorflow/python/keras/ |
| metrics.py | 139 self._dtype = K.floatx() if dtype is None else dtypes.as_dtype(dtype).name
188 return self._dtype
247 dtype=self._dtype if dtype is None else dtype,
291 values = math_ops.cast(values, self._dtype)
293 sample_weight = math_ops.cast(sample_weight, self._dtype)
323 num_values = math_ops.cast(array_ops.size(values), self._dtype)
326 num_values = math_ops.cast(array_ops.size(values), self._dtype)
476 normalizer = math_ops.cast(normalizer, self._dtype)
492 y_true = math_ops.cast(y_true, self._dtype)
493 y_pred = math_ops.cast(y_pred, self._dtype)
[all...] |
| metrics_test.py | 480 self.assertEqual(cosine_obj._dtype, dtypes.int32)
485 self.assertEqual(cosine_obj2._dtype, dtypes.int32)
523 self.assertEqual(mae_obj._dtype, dtypes.int32)
528 self.assertEqual(mae_obj2._dtype, dtypes.int32)
562 self.assertEqual(mape_obj._dtype, dtypes.int32)
568 self.assertEqual(mape_obj2._dtype, dtypes.int32)
601 self.assertEqual(mse_obj._dtype, dtypes.int32)
606 self.assertEqual(mse_obj2._dtype, dtypes.int32)
640 self.assertEqual(msle_obj._dtype, dtypes.int32)
646 self.assertEqual(msle_obj2._dtype, dtypes.int32
[all...] |
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| math_utils.py | 703 self._dtype = dtype
740 num_features=self._num_features, dtype=self._dtype)
[all...] |
| /external/tensorflow/tensorflow/contrib/slim/python/slim/data/ |
| tfexample_decoder.py | 389 self._dtype = dtype
400 image_buffer, dtype=self._dtype)
422 self._dtype)
430 dct_method=self._dct_method), self._dtype)
444 return parsing_ops.decode_raw(image_buffer, out_type=self._dtype)
|
| /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
| io_ops.py | 45 self._dtype = dtype
54 return self._dtype
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| xla_client.py | 507 self._dtype = dtype
514 return (self._dtype == other._dtype and
522 return hash((self._dtype, self._dimensions, self._minor_to_major))
525 return ('xla_client.Shape(_dtype={!r}, _dimensions={!r}, '
527 self._dtype, self._dimensions, self._is_tuple,
554 return self._dtype
[all...] |
| /external/tensorflow/tensorflow/contrib/tpu/python/tpu/ |
| keras_tpu_variables.py | 75 self._dtype = variables[0].dtype
149 v = gen_resource_variable_ops.read_variable_op(self.handle, self._dtype)
|
| /external/tensorflow/tensorflow/contrib/eager/python/ |
| parameter_server.py | 203 self._dtype = initial_value.dtype.base_dtype
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| base_layer.py | 329 if self._dtype is None:
330 self._dtype = dtype.base_dtype.name
669 return self._dtype
[all...] |
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| linear_operator.py | 202 self._dtype = dtype
220 return self._dtype
[all...] |
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| bijector_impl.py | 569 self._dtype = dtype
637 return self._dtype
[all...] |
| distribution.py | 471 self._dtype = dtype
569 return self._dtype
[all...] |
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| ragged_tensor.py | [all...] |
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| cudnn_rnn_test.py | 98 self._dtype = dtype
172 np_dtype = np.float32 if self._dtype == dtypes.float32 else np.float64
190 np_dtype = np.float32 if self._dtype == dtypes.float32 else np.float64
[all...] |
| /external/tensorflow/tensorflow/python/eager/ |
| def_function.py | 153 self._dtype = initial_value.dtype.base_dtype
[all...] |
