Searched refs:t_dtype (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_ex_op_test.cc | 236 DataType t_dtype = DataTypeToEnum<T>::v(); in RunFusedBatchNormEx() local
262 side_inputs.push_back({side_input.name(), 0, t_dtype}); in RunFusedBatchNormEx()
267 .Input({input.name(), 0, t_dtype}) in RunFusedBatchNormEx()
273 .Attr("T", t_dtype) in RunFusedBatchNormEx()
285 .Input({y_backprop.name(), 0, t_dtype}) in RunFusedBatchNormEx()
286 .Input({fused_batch_norm_ex.name(), 0, t_dtype}) in RunFusedBatchNormEx()
287 .Attr("T", t_dtype) in RunFusedBatchNormEx()
291 .Input({y_backprop.name(), 0, t_dtype}) in RunFusedBatchNormEx()
292 .Attr("T", t_dtype) in RunFusedBatchNormEx()
298 .Input({activation_grad.name(), 0, t_dtype}) in RunFusedBatchNormEx()
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_util.py | 800 t_dtype = default_gradient.get_zeros_dtype(t)
803 resource_variable_ops.variable_shape(t), dtype=t_dtype)
805 return array_ops.zeros_like(t, dtype=t_dtype)
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper.cc | 853 DataType t_dtype = GetDataTypeFromAttr(*fused_batch_norm_node_def, "T"); in IsCpuCompatibleDataType() local
855 if (t_dtype != DT_FLOAT && t_dtype != DT_HALF) return false; in IsCpuCompatibleDataType()
857 if (t_dtype != DT_FLOAT && t_dtype != DT_BFLOAT16) return false; in IsCpuCompatibleDataType()
877 if (t_dtype != DT_HALF) return false; in IsCpuCompatibleDataType()
|