Home
last modified time | relevance | path

Searched refs:base_dtype (Results 1 – 25 of 99) sorted by relevance

1234

/external/tensorflow/tensorflow/python/training/
Dftrl.py171 math_ops.cast(self._learning_rate_tensor, var.dtype.base_dtype),
173 var.dtype.base_dtype),
175 var.dtype.base_dtype),
176 math_ops.cast(self._learning_rate_power_tensor, var.dtype.base_dtype),
184 math_ops.cast(self._learning_rate_tensor, var.dtype.base_dtype),
186 var.dtype.base_dtype),
188 var.dtype.base_dtype),
190 var.dtype.base_dtype),
191 math_ops.cast(self._learning_rate_power_tensor, var.dtype.base_dtype),
203 math_ops.cast(self._learning_rate_tensor, var.dtype.base_dtype),
[all …]
Drmsprop.py123 init_rms = init_ops.ones_initializer(dtype=v.dtype.base_dtype)
127 v.dtype.base_dtype, "rms",
154 math_ops.cast(self._learning_rate_tensor, var.dtype.base_dtype),
155 math_ops.cast(self._decay_tensor, var.dtype.base_dtype),
156 math_ops.cast(self._momentum_tensor, var.dtype.base_dtype),
157 math_ops.cast(self._epsilon_tensor, var.dtype.base_dtype),
165 math_ops.cast(self._learning_rate_tensor, var.dtype.base_dtype),
166 math_ops.cast(self._decay_tensor, var.dtype.base_dtype),
167 math_ops.cast(self._momentum_tensor, var.dtype.base_dtype),
168 math_ops.cast(self._epsilon_tensor, var.dtype.base_dtype),
[all …]
Dadam.py160 math_ops.cast(beta1_power, var.dtype.base_dtype),
161 math_ops.cast(beta2_power, var.dtype.base_dtype),
162 math_ops.cast(self._lr_t, var.dtype.base_dtype),
163 math_ops.cast(self._beta1_t, var.dtype.base_dtype),
164 math_ops.cast(self._beta2_t, var.dtype.base_dtype),
165 math_ops.cast(self._epsilon_t, var.dtype.base_dtype),
177 math_ops.cast(beta1_power, grad.dtype.base_dtype),
178 math_ops.cast(beta2_power, grad.dtype.base_dtype),
179 math_ops.cast(self._lr_t, grad.dtype.base_dtype),
180 math_ops.cast(self._beta1_t, grad.dtype.base_dtype),
[all …]
Dadagrad_da.py94 0.0, shape=v.get_shape(), dtype=v.dtype.base_dtype)
98 dtype=v.dtype.base_dtype)
121 math_ops.cast(self._learning_rate_tensor, var.dtype.base_dtype),
122 math_ops.cast(self._l1_regularization_strength, var.dtype.base_dtype),
123 math_ops.cast(self._l2_regularization_strength, var.dtype.base_dtype),
137 math_ops.cast(self._learning_rate_tensor, grad.dtype.base_dtype),
138 math_ops.cast(self._l1_regularization_strength, grad.dtype.base_dtype),
139 math_ops.cast(self._l2_regularization_strength, grad.dtype.base_dtype),
154 math_ops.cast(self._learning_rate_tensor, var.dtype.base_dtype),
155 math_ops.cast(self._l1_regularization_strength, var.dtype.base_dtype),
[all …]
Dadadelta.py90 math_ops.cast(self._lr_t, var.dtype.base_dtype),
91 math_ops.cast(self._rho_t, var.dtype.base_dtype),
92 math_ops.cast(self._epsilon_t, var.dtype.base_dtype),
103 math_ops.cast(self._lr_t, grad.dtype.base_dtype),
104 math_ops.cast(self._rho_t, grad.dtype.base_dtype),
105 math_ops.cast(self._epsilon_t, grad.dtype.base_dtype),
116 math_ops.cast(self._lr_t, var.dtype.base_dtype),
117 math_ops.cast(self._rho_t, var.dtype.base_dtype),
118 math_ops.cast(self._epsilon_t, var.dtype.base_dtype),
Dmomentum.py104 math_ops.cast(self._learning_rate_tensor, var.dtype.base_dtype),
106 math_ops.cast(self._momentum_tensor, var.dtype.base_dtype),
114 math_ops.cast(self._learning_rate_tensor, grad.dtype.base_dtype),
116 math_ops.cast(self._momentum_tensor, grad.dtype.base_dtype),
124 math_ops.cast(self._learning_rate_tensor, var.dtype.base_dtype),
126 math_ops.cast(self._momentum_tensor, var.dtype.base_dtype),
Dslot_creator_test.py48 self.assertEqual(dtypes.float32, slot.dtype.base_dtype)
61 self.assertEqual(dtypes.float32, slot.dtype.base_dtype)
76 self.assertEqual(dtypes.float64, slot.dtype.base_dtype)
98 self.assertEqual(dtypes.float64, slot.dtype.base_dtype)
112 self.assertEqual(dtypes.float32, slot.dtype.base_dtype)
128 self.assertEqual(dtypes.float64, slot.dtype.base_dtype)
156 self.assertEqual(dtypes.float32, slot.dtype.base_dtype)
177 self.assertEqual(dtypes.float32, slot.dtype.base_dtype)
Dadagrad.py74 dtype = v.dtype.base_dtype
102 math_ops.cast(self._learning_rate_tensor, var.dtype.base_dtype),
111 math_ops.cast(self._learning_rate_tensor, grad.dtype.base_dtype),
120 math_ops.cast(self._learning_rate_tensor, var.dtype.base_dtype),
Dgradient_descent.py58 math_ops.cast(self._learning_rate_tensor, var.dtype.base_dtype),
65 grad.dtype.base_dtype),
75 math_ops.cast(self._learning_rate_tensor, var.dtype.base_dtype),
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Dlegacy_learning_rate_decay.py159 if b.dtype.base_dtype != x_recomp.dtype.base_dtype:
163 if (b.dtype.base_dtype == dtypes.int32 and
164 x_recomp.dtype.base_dtype == dtypes.int64):
165 b = math_ops.cast(b, x_recomp.dtype.base_dtype)
170 (b.dtype.base_dtype, x_recomp.dtype.base_dtype))
172 if v.dtype.base_dtype != values[0].dtype.base_dtype:
175 (values[0].dtype.base_dtype, v.dtype.base_dtype))
Dadagrad.py81 dtype = var.dtype.base_dtype
128 var_device, var_dtype = var.device, var.dtype.base_dtype
142 var_device, var_dtype = var.device, var.dtype.base_dtype
Dgradient_descent.py133 var_device, var_dtype = var.device, var.dtype.base_dtype
160 var_device, var_dtype = var.device, var.dtype.base_dtype
171 var_device, var_dtype = var.device, var.dtype.base_dtype
Dftrl.py133 dtype = var.dtype.base_dtype
154 var_device, var_dtype = var.device, var.dtype.base_dtype
192 var_device, var_dtype = var.device, var.dtype.base_dtype
Dnadam.py91 var_dtype = var_list[0].dtype.base_dtype
151 var_device, var_dtype = var.device, var.dtype.base_dtype
174 var_device, var_dtype = var.device, var.dtype.base_dtype
Dadadelta.py118 var_device, var_dtype = var.device, var.dtype.base_dtype
135 var_device, var_dtype = var.device, var.dtype.base_dtype
/external/tensorflow/tensorflow/python/ops/distributions/
Dcategorical.py36 def _broadcast_cat_event_and_params(event, params, base_dtype): argument
46 "`self.dtype` ({})".format(base_dtype))
291 k, self.probs, base_dtype=self.dtype.base_dtype)
312 k, self.logits, base_dtype=self.dtype.base_dtype)
/external/tensorflow/tensorflow/python/ops/
Dsets_impl.py56 if a.values.dtype.base_dtype not in _VALID_DTYPES:
84 if a.dtype.base_dtype not in _VALID_DTYPES:
87 if b.dtype.base_dtype != a.dtype.base_dtype:
Dmath_ops.py477 dtype = y.dtype.base_dtype if tensor_util.is_tf_type(y) else None
612 scalar, dtype=x.dtype.base_dtype, name="scalar")
941 base_type = dtypes.as_dtype(dtype).base_dtype
958 if x.dtype.base_dtype != base_type:
986 dtype = dtypes.as_dtype(dtype).base_dtype
1168 dtype = tensors[0].dtype.base_dtype
1225 y = ops.convert_to_tensor(y, dtype=sp_x.dtype.base_dtype, name="y")
1284 x_dtype = sp_values.dtype.base_dtype
1285 y_dtype = y.dtype.base_dtype
1303 y = ops.convert_to_tensor(y, dtype_hint=x.dtype.base_dtype, name="y")
[all …]
Dpartitioned_variables.py132 if dtype.base_dtype == dtypes.string:
205 if dtype.base_dtype == dtypes.string:
Dlookup_ops.py101 if key_dtype.base_dtype != table.key_dtype:
104 if value_dtype.base_dtype != table.value_dtype:
229 if keys.dtype.base_dtype != self._key_dtype:
924 if dtypes.string == tensor.dtype.base_dtype:
1112 if keys.dtype.base_dtype != self._key_dtype:
1119 if self._table and (self._table.key_dtype.base_dtype == dtypes.int64):
1310 if keys.dtype.base_dtype != self._key_dtype:
1317 if self._table and (self._table.key_dtype.base_dtype == dtypes.int64):
1446 if (not key_dtype.is_integer) and (dtypes.string != key_dtype.base_dtype):
1538 if (not dtype.is_integer) and (dtypes.string != dtype.base_dtype):
[all …]
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator_util.py111 dtype_base = base_dtype(dtype)
112 value_dtype_base = base_dtype(value.dtype)
121 def base_dtype(dtype): function
125 return dtype.base_dtype
141 if arg.dtype.base_dtype != dtype:
193 dtype = x.dtype.base_dtype
212 dtype = x.dtype.base_dtype
/external/tensorflow/tensorflow/python/tpu/
Dbfloat16_test.py77 self.assertEqual(v1.dtype.base_dtype, dtypes.float32)
79 self.assertEqual(v2.dtype.base_dtype, dtypes.bfloat16)
81 [v.dtype.base_dtype for v in scope.global_variables()])
/external/tensorflow/tensorflow/python/eager/
Dbackprop_util.py31 if dtype.base_dtype == dtypes.variant:
59 return dtype.base_dtype in (dtypes.float16, dtypes.float32, dtypes.float64,
/external/tensorflow/tensorflow/python/keras/layers/ops/
Dcore.py45 if inputs.dtype.base_dtype != dtype.base_dtype:
/external/tensorflow/tensorflow/python/framework/
Ddtypes.py85 def base_dtype(self): member in DType
95 base = self.base_dtype
117 self.base_dtype in (bool, string, complex64, complex128)):
128 if self.base_dtype == bfloat16:
141 self.base_dtype in (bool, string, complex64, complex128)):
152 if self.base_dtype == bfloat16:
190 other.base_dtype.as_datatype_enum)

1234