可用的 TensorFlow 運算

本頁列出 Cloud TPU 可用的 TensorFlow Python API 及圖形運算子。

可用的 Python API

下方清單列出可用的 TensorFlow Python API 組合。本清單並非詳盡無遺。不在這份清單的資料庫函式，如果是以可用基元組成，就可能可以使用。

請參閱效能指南，瞭解有關特定運算子的建議內容。

模組	可用的 Python API	註解
tf	tf.abs
	tf.acosh
	tf.add
	tf.add_n
	tf.angle
	tf.arg_max	dimension 引數必須是編譯時間常數。
	tf.arg_min	dimension 引數必須是編譯時間常數。
	tf.asinh
	tf.assign	僅提供給資源變數。
	tf.assign_add	僅提供給資源變數。
	tf.assign_sub	僅提供給資源變數。
	tf.atan
	tf.atan2
	tf.atanh
	tf.batch_to_space	crops 和 block_shape 引數必須是編譯時常數。
	tf.batch_to_space_nd	crops 引數必須是編譯時間常數。
	tf.broadcast_dynamic_shape
	tf.broadcast_static_shape
	tf.case	實驗版 (控制流程)。可能尚無法可靠運作。
	tf.cast
	tf.ceil
	tf.cholesky	實驗版。可能有數值精度問題。
	tf.cholesky_solve	實驗版。可能有數值精度問題。
	tf.clip_by_average_norm
	tf.clip_by_global_norm
	tf.clip_by_norm
	tf.clip_by_value
	tf.complex
	tf.concat	concat_dim 必須是編譯時間常數。
	tf.cond	實驗版 (控制流程)。可能尚無法可靠運作。
	tf.conj
	tf.constant
	tf.convert_to_tensor
	tf.cos
	tf.cosh
	tf.cross
	tf.cumprod	axis 必須是編譯時間常數。
	tf.cumsum	axis 必須是編譯時間常數。
	tf.depth_to_space
	tf.diag
	tf.diag_part
	tf.div	int32 除法比其他類型的運算速度較慢。
	tf.divide	int32 除法比其他類型的運算速度較慢。
	tf.dynamic_stitch	indices 必須是編譯時間常數。
	tf.einsum
	tf.equal
	tf.erf
	tf.erfc
	tf.exp
	tf.expand_dims	dims 必須是編譯時間常數。
	tf.expm1
	tf.extract_image_patches
	tf.eye
	tf.fake_quant_with_min_max_args
	tf.fake_quant_with_min_max_args_gradient
	tf.fake_quant_with_min_max_vars
	tf.fake_quant_with_min_max_vars_gradient
	tf.fft
	tf.fft2d
	tf.fft3d
	tf.fill	dims 引數必須是編譯時間常數。
	tf.floor
	tf.floordiv
	tf.floormod
	tf.foldl	實驗版 (控制流程)。
	tf.foldr	實驗版 (控制流程)。
	tf.gather	axis 必須是編譯時間常數。
	tf.gather_nd
	tf.greater
	tf.greater_equal
	tf.hessians	實驗版 (控制流程)。
	tf.identity
	tf.identity_n
	tf.ifft
	tf.ifft2d
	tf.ifft3d
	tf.imag
	tf.invert_permutation	x 引數必須是編譯時間常數。
	tf.is_finite
	tf.is_inf
	tf.is_nan
	tf.is_non_decreasing
	tf.is_strictly_increasing
	tf.less
	tf.less_equal
	tf.linspace	start、stop 和 num 引數必須是編譯時間常數。
	tf.log
	tf.log1p
	tf.log_sigmoid
	tf.logical_and
	tf.logical_or
	tf.logical_not
	tf.logical_xor
	tf.matmul	使用 bfloat16 與 float32 累加的 matmul。
	tf.matrix_band_part
	tf.matrix_diag
	tf.matrix_diag_part
	tf.matrix_set_diag
	tf.matrix_triangular_solve	實驗版。可能有數值精度問題。
	tf.maximum
	tf.meshgrid
	tf.minimum
	tf.mod
	tf.multinomial	num_samples 引數必須是編譯時間常數。
	tf.multiply
	tf.negative
	tf.no_op
	tf.norm
	tf.not_equal
	tf.one_hot	depth 必須是編譯時間常數。
	tf.ones
	tf.ones_like
	tf.pad	paddings 引數必須是編譯時常數。REFLECT 邊框的漸層效果尚未推出。
	tf.pow
	tf.random_normal	shape 必須是編譯時間常數。
	tf.random_uniform	shape 必須是編譯時間常數。
	tf.range	start、limit 和 delta 引數必須是編譯時間常數。
	tf.rank
	tf.real
	tf.realdiv
	tf.reciprocal
	tf.reduce_all	axis 必須是編譯時間常數。
	tf.reduce_any	axis 必須是編譯時間常數。
	tf.reduce_logsumexp
	tf.reduce_max	axis 必須是編譯時間常數。
	tf.reduce_min	axis 必須是編譯時間常數。
	tf.reduce_prod	axis 必須是編譯時間常數。
	tf.reduce_sum	axis 必須是編譯時間常數。
	tf.reshape	shape 引數必須是編譯時間常數。
	tf.reverse	dims 引數必須是編譯時間常數。
	tf.reverse_sequence
	tf.reverse_v2	axis 引數必須是編譯時間常數。
	tf.rint
	tf.round
	tf.rsqrt
	tf.saturate_cast
	tf.scalar_mul
	tf.scan	實驗版 (控制流程)。
	tf.scatter_nd
	tf.sequence_mask
	tf.shape
	tf.shape_n
	tf.sigmoid
	tf.sign
	tf.sin
	tf.sinh
	tf.size
	tf.slice	size 必須是編譯時間常數。此外，begin 必須是編譯時常數，或是 size 必須為非負值。只有在 begin 和 size 為編譯時間常數時，才支援回傳。
	tf.space_to_batch	paddings 和 block_shape 必須是編譯時間常數。
	tf.space_to_batch_nd	paddings 必須是編譯時間常數。
	tf.space_to_depth
	tf.split	axis 必須是編譯時間常數。
	tf.sqrt
	tf.square
	tf.squared_difference
	tf.squeeze
	tf.stack
	tf.stop_gradient
	tf.strided_slice
	tf.tan
	tf.tanh
	tf.tensordot
	tf.tile	multiples 引數必須是編譯時間常數。
	tf.to_bfloat16
	tf.to_float
	tf.to_int32
	tf.to_int64	int64 支援功能有限。
	tf.trace
	tf.transpose	perm 引數必須是編譯時間常數。
	tf.truediv
	tf.truncated_normal	shape 必須是編譯時間常數。
	tf.truncatediv
	tf.truncatemod
	tf.unsorted_segment_sum
	tf.unstack
	tf.where	x 和 y 都必須是非 None。如果 x 和 y 都是 None，運算子就不會具有靜態形狀。
	tf.while_loop	計算 while 迴圈的梯度時，必須傳遞 maximum_iterations 引數。
	tf.zeros
	tf.zeros_like
	tf.Tensor.__getitem__	配量的開始、結束及步距必須為編譯時間常數。
tf.bitwise	tf.bitwise_and
	tf.bitwise_or
	tf.bitwise_invert
tf.contrib.stateless	tf.contrib.stateless.stateless_random_normal
	tf.contrib.stateless.stateless_random_uniform
tf.image	tf.image.adjust_brightness
	tf.image.adjust_contrast
	tf.image.adjust_gamma
	tf.image.adjust_hue
	tf.image.adjust_saturation
	tf.image.central_crop	裁剪因數必須為編譯時間常數。
	tf.image.convert_image_dtype
	tf.image.flip_left_right
	tf.image.flip_up_down
	tf.image.grayscale_to_rgb
	tf.image.hsv_to_rgb
	tf.image.resize_bilinear	僅提供 align_corners=True。size 必須是編譯時間常數。
	tf.image.random_brightness
	tf.image.random_contrast
	tf.image.random_flip_left_right
	tf.image.random_flip_up_down
	tf.image.random_hue
	tf.image.random_saturation
	tf.image.rgb_to_hsv
	tf.image.rgb_to_grayscale
	tf.image.rot90
	tf.image.total_variation
	tf.image.transpose_image
tf.layers	tf.layers.average_pooling1d
	tf.layers.average_pooling2d
	tf.layers.average_pooling1d
	tf.layers.batch_normalization
	tf.layers.conv1d
	tf.layers.conv2d
	tf.layers.conv2d_transpose
	tf.layers.conv3d
	tf.layers.conv3d_transpose
	tf.layers.dense
	tf.layers.dropout
	tf.layers.flatten
	tf.layers.max_pooling1d
	tf.layers.max_pooling2d
	tf.layers.max_pooling3d
	tf.layers.separable_conv2d
tf.nn	tf.nn.atrous_conv2d
	tf.nn.atrous_conv2d_transpose
	tf.nn.avg_pool
	tf.nn.avg_pool3d
	tf.nn.batch_normalization
	tf.nn.bias_add
	tf.nn.conv1d
	tf.nn.conv2d
	tf.nn.conv2d_backprop_filter
	tf.nn.conv2d_backprop_input
	tf.nn.conv2d_transpose
	tf.nn.conv3d
	tf.nn.conv3d_backprop_filter
	tf.nn.conv3d_backprop_input
	tf.nn.conv3d_transpose
	tf.nn.convolution
	tf.nn.crelu
	tf.nn.depthwise_conv2d
	tf.nn.depthwise_conv2d_native
	tf.nn.depthwise_conv2d_native_backprop_filter
	tf.nn.depthwise_conv2d_native_backprop_input
	tf.nn.dropout
	tf.nn.dynamic_rnn	實驗版。
	tf.nn.elu
	tf.nn.fused_batch_norm
	tf.nn.l2_loss
	tf.nn.l2_normalize
	tf.nn.leaky_relu
	tf.nn.local_response_normalization
	tf.nn.log_poisson_loss
	tf.nn.log_softmax
	tf.nn.max_pool
	tf.nn.max_pool3d
	tf.nn.moments
	tf.nn.normalize_moments
	tf.nn.pool
	tf.nn.relu
	tf.nn.relu6
	tf.nn.relu_layer
	tf.nn.selu
	tf.nn.separable_conv2d
	tf.nn.sigmoid_cross_entropy_with_logits
	tf.nn.softmax
	tf.nn.softmax_cross_entropy_with_logits
	tf.nn.softplus
	tf.nn.softsign
	tf.nn.sparse_softmax_cross_entropy_with_logits
	tf.nn.static_bidirectional_rnn	實驗版。
	tf.nn.static_rnn	實驗版。
	tf.nn.weighted_cross_entropy_with_logits	實驗版。
	tf.nn.weighted_moments
	tf.nn.with_space_to_batch
	tf.nn.xw_plus_b
	tf.nn.zero_fraction
tf.spectral	tf.spectral.fft
	tf.spectral.fft2d
	tf.spectral.fft3d
	tf.spectral.ifft
	tf.spectral.ifft2d
	tf.spectral.ifft3d
	tf.spectral.irfft	fft_length 必須是編譯時間常數。
	tf.spectral.irfft2d	fft_length 必須是編譯時間常數。
	tf.spectral.irfft3d	fft_length 必須是編譯時間常數。
	tf.spectral.rfft	fft_length 必須是編譯時間常數。
	tf.spectral.rfft2d	fft_length 必須是編譯時間常數。
	tf.spectral.rfft3d	fft_length 必須是編譯時間常數。

不可用的 Python API

本清單並非詳盡無遺。以下是無法在 Cloud TPU 使用的運算：

模組	不可用的 Python API	註解
tf	tf.accumulate_n	使用 Ref 變數。
	tf.acos
	tf.asin
	tf.betainc
	tf.bitcast
	tf.add_check_numerics_ops	含有判斷數值運算子的程式應能執行，但判斷數值運算子目前將遭到忽略。
	tf.assert_...	含有判斷提示的程式應能執行，但判斷提示將遭到忽略。
	tf.check_numerics	含有判斷數值運算子的程式應能執行，但判斷數值運算子目前將遭到忽略。
	tf.confusion_matrix
	tf.count_nonzero	使用 int64 縮減功能。
	tf.count_up_to
	tf.create_partitioned_variables
	tf.dequantize
	tf.digamma
	tf.dynamic_partition
	tf.edit_distance
	tf.fake_quant_with_min_max_vars_per_channel
	tf.fake_quant_with_min_max_vars_per_channel_gradient
	tf.histogram_fixed_width
	tf.igamma
	tf.igammac
	tf.lbeta
	tf.lgamma
	tf.matrix_determinant
	tf.matrix_inverse
	tf.matrix_solve
	tf.matrix_solve_ls
	tf.polygamma
	tf.py_func
	tf.qr
	tf.quantize_v2
	tf.quantized_concat
	tf.random_crop
	tf.random_gamma
	tf.random_poisson
	tf.random_shuffle
	tf.scatter_add
	tf.scatter_div
	tf.scatter_mul
	tf.scatter_nd_add
	tf.scatter_nd_sub
	tf.scatter_nd_update
	tf.segment_mean
	tf.segment_max
	tf.segment_min
	tf.segment_prod
	tf.segment_sum
	tf.self_adjoint_eig
	tf.self_adjoint_eigvals
	tf.setdiff1d
	tf.sparse_...
	tf.string_...
	tf.substr
	tf.svd
	tf.to_double
	tf.unique
	tf.unsorted_segment_max
	tf.zeta
	tf.bitwise.bitwise_xor
	tf.contrib.stateless.stateless_truncated_normal

可用的圖形運算子

運算子	類型條件約束
Abs	T={bfloat16,float,int32,int64}
Acos	T={bfloat16,complex64,float,int32,int64}
Acosh	T={bfloat16,complex64,float}
Add	T={bfloat16,complex64,float,int32,int64}
AddN	T={bfloat16,complex64,float,int32,int64,uint32,uint64}
AdjustContrastv2	T={float}
AdjustHue	T={float}
AdjustSaturation	T={float}
All	Tidx={int32,int64}
AllToAll	T={bfloat16,float}
Angle	Tout={float} T={complex64}
Any	Tidx={int32,int64}
ApproximateEqual	T={bfloat16,complex64,float,int32,int64,uint32,uint64}
ArgMax	Tidx={int32,int64} output_type={int32,int64} T={bfloat16,complex64,float,int32,int64,uint32,uint64}
ArgMin	Tidx={int32,int64} output_type={int32,int64} T={bfloat16,complex64,float,int32,int64,uint32,uint64}
Asin	T={bfloat16,complex64,float,int32,int64}
Asinh	T={bfloat16,complex64,float}
Assert	T={bfloat16,bool,complex64,float,int32,int64,string,uint32,uint64}
AssignAddVariableOp	dtype={bfloat16,complex64,float,int32,int64,uint32,uint64}
AssignSubVariableOp	dtype={bfloat16,complex64,float,int32,int64,uint32,uint64}
AssignVariableOp	dtype={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
Atan	T={bfloat16,complex64,float,int32,int64}
Atan2	T={bfloat16,float}
Atanh	T={bfloat16,complex64,float}
AvgPool	T={bfloat16,float}
AvgPool3D	T={bfloat16,float}
AvgPool3DGrad	T={bfloat16,float}
AvgPoolGrad	T={bfloat16,float}
BatchMatMul	T={bfloat16,complex64,float,int32,int64}
BatchToSpace	Tidx={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
BatchToSpaceND	Tcrops={int32,int64} Tblock_shape={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
BiasAdd	T={bfloat16,complex64,float,int32,int64,uint32,uint64}
BiasAddGrad	T={bfloat16,complex64,float,int32,int64,uint32,uint64}
BiasAddV1	T={bfloat16,complex64,float,int32,int64,uint32,uint64}
Bitcast	type={bfloat16,complex64,float,int32,int64,uint32,uint64} T={bfloat16,complex64,float,int32,int64,uint32,uint64}
BitwiseAnd	T={int32,int64,uint32,uint64}
BitwiseOr	T={int32,int64,uint32,uint64}
BitwiseXor	T={int32,int64,uint32,uint64}
BroadcastArgs	T={int32,int64}
BroadcastGradientArgs	T={int32,int64}
BroadcastTo	Tidx={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
Bucketize	T={float,int32,int64}
Cast	DstT={bfloat16,bool,complex64,float,int32,int64,uint32,uint64} SrcT={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
Ceil	T={bfloat16,float}
CheckNumerics	T={bfloat16,float}
Cholesky	T={float}
ClipByValue	T={bfloat16,complex64,float,int32,int64,uint32,uint64}
CollectivePermute	T={bfloat16,float}
Complex	Tout={complex64} T={float}
ComplexAbs	Tout={float} T={complex64}
Concat	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
ConcatOffset
ConcatV2	Tidx={int32} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
Conj	T={complex64}
ConjugateTranspose	Tperm={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
Const	dtype={bfloat16,bool,complex64,float,int32,int64,string,uint32,uint64}
ControlTrigger
Conv2D	T={bfloat16,float}
Conv2DBackpropFilter	T={bfloat16,float}
Conv2DBackpropInput	T={bfloat16,float}
Conv3D	T={bfloat16,float}
Conv3DBackpropFilterV2	T={bfloat16,float}
Conv3DBackpropInputV2	Tshape={int32,int64} T={bfloat16,float}
Cos	T={bfloat16,complex64,float}
Cosh	T={bfloat16,complex64,float}
Cross	T={bfloat16,float,int32,int64,uint32,uint64}
CrossReplicaSum	T={bfloat16,float}
Cumprod	Tidx={int32,int64} T={bfloat16,float,int32}
Cumsum	Tidx={int32,int64} T={bfloat16,float,int32}
DataFormatVecPermute	T={int32,int64}
DepthToSpace	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
DepthwiseConv2dNative	T={bfloat16,float}
DepthwiseConv2dNativeBackpropFilter	T={bfloat16,float}
DepthwiseConv2dNativeBackpropInput	T={bfloat16,float}
Diag	T={bfloat16,complex64,float,int32,int64}
DiagPart	T={bfloat16,complex64,float,int32,int64}
Digamma	T={bfloat16,float}
Div	T={bfloat16,complex64,float,int32,int64}
DivNoNan	T={float}
DynamicStitch	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
Elu	T={bfloat16,float}
EluGrad	T={bfloat16,float}
Empty	dtype={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
EmptyTensorList	shape_type={int32,int64} element_dtype={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
Equal	T={bfloat16,bool,complex64,float,int32,int64}
Erf	T={bfloat16,float}
Erfc	T={bfloat16,float}
Exp	T={bfloat16,complex64,float}
ExpandDims	Tdim={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
Expm1	T={bfloat16,complex64,float}
ExtractImagePatches	T={bfloat16,float,int32,int64,uint32,uint64}
FFT	Tcomplex={complex64}
FFT2D	Tcomplex={complex64}
FFT3D	Tcomplex={complex64}
FakeParam	dtype={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
FakeQuantWithMinMaxArgs
FakeQuantWithMinMaxArgsGradient
FakeQuantWithMinMaxVars
FakeQuantWithMinMaxVarsGradient
Fill	index_type={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
Floor	T={bfloat16,float}
FloorDiv	T={bfloat16,complex64,float,int32,int64}
FloorMod	T={bfloat16,float,int32,int64}
FusedBatchNorm	T={float}
FusedBatchNormGrad	T={float}
FusedBatchNormGradV2	U={float} T={bfloat16,float}
FusedBatchNormV2	U={float} T={bfloat16,float}
Gather	Tindices={int32,int64} Tparams={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
GatherNd	Tindices={int32,int64} Tparams={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
GatherV2	Taxis={int32,int64} Tindices={int32,int64} Tparams={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
GetItem	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
Greater	T={bfloat16,float,int32,int64,uint32,uint64}
GreaterEqual	T={bfloat16,float,int32,int64,uint32,uint64}
HSVToRGB	T={bfloat16,float}
IFFT	Tcomplex={complex64}
IFFT2D	Tcomplex={complex64}
IFFT3D	Tcomplex={complex64}
IRFFT
IRFFT2D
IRFFT3D
Identity	T={bfloat16,bool,complex64,float,int32,int64,resource,uint32,uint64}
IdentityN	T={bfloat16,bool,complex64,float,int32,int64,resource,uint32,uint64}
If	Tout={bfloat16,bool,complex64,float,int32,int64,resource,uint32,uint64} Tin={bfloat16,bool,complex64,float,int32,int64,resource,uint32,uint64} Tcond={bfloat16,bool,complex64,float,int32,int64,resource,uint32,uint64}
Imag	Tout={float} T={complex64}
InfeedDequeue	dtype={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
InfeedDequeueTuple	dtypes={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
InplaceAdd	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
InplaceUpdate	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
Inv	T={bfloat16,complex64,float,int32,int64}
Invert	T={int32,int64,uint32,uint64}
InvertPermutation	T={int32}
IsFinite	T={bfloat16,float}
IsInf	T={bfloat16,float}
IsNan	T={bfloat16,float}
L2Loss	T={bfloat16,float}
LRN	T={bfloat16,float}
LRNGrad	T={bfloat16,float}
LeakyRelu	T={bfloat16,float}
LeakyReluGrad	T={bfloat16,float}
LeftShift	T={int32,int64,uint32,uint64}
Less	T={bfloat16,float,int32,int64,uint32,uint64}
LessEqual	T={bfloat16,float,int32,int64,uint32,uint64}
Lgamma	T={bfloat16,float}
LinSpace	Tidx={int32,int64} T={bfloat16,float}
ListDiff	out_idx={int32,int64} T={int32,int64}
Log	T={bfloat16,complex64,float}
Log1p	T={bfloat16,complex64,float}
LogSoftmax	T={bfloat16,float}
LogicalAnd
LogicalNot
LogicalOr
MatMul	T={bfloat16,complex64,float}
MatrixBandPart	Tindex={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
MatrixDiag	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
MatrixDiagPart	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
MatrixSetDiag	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
MatrixTriangularSolve	T={complex64,float}
Max	Tidx={int32,int64} T={bfloat16,complex64,float,int32,int64,uint32,uint64}
MaxPool	T={bfloat16,float,int32,int64}
MaxPool3D	T={bfloat16,float}
MaxPool3DGrad	TInput={bfloat16,float} T={bfloat16,float}
MaxPool3DGradGrad	T={float}
MaxPoolGrad	T={bfloat16,float,int32,int64,uint32,uint64}
MaxPoolGradGrad	T={float}
MaxPoolGradGradV2	T={float}
MaxPoolGradV2	T={bfloat16,float,int32,int64,uint32,uint64}
MaxPoolV2	T={bfloat16,float,int32,int64}
Maximum	T={bfloat16,float,int32,int64}
Mean	Tidx={int32,int64} T={bfloat16,complex64,float,int32,int64,uint32,uint64}
Min	Tidx={int32,int64} T={bfloat16,complex64,float,int32,int64,uint32,uint64}
Minimum	T={bfloat16,float,int32,int64}
MirrorPad	Tpaddings={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
Mod	T={bfloat16,float,int32,int64}
Mul	T={bfloat16,complex64,float,int32,int64}
Multinomial	output_dtype={int32,int64} T={bfloat16,float,int32,int64,uint32,uint64}
Neg	T={bfloat16,complex64,float,int32,int64}
NoOp
NonMaxSuppressionV4	T={float}
NotEqual	T={bfloat16,bool,complex64,float,int32,int64}
OneHot	TI={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
OnesLike	T={bfloat16,bool,complex64,float,int32,int64}
OutfeedEnqueue	dtype={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
OutfeedEnqueueTuple	dtypes={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
Pack	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
Pad	Tpaddings={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
PadV2	Tpaddings={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
ParallelDynamicStitch	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
PlaceholderWithDefault	dtype={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
Pow	T={bfloat16,complex64,float,int32,int64}
PreventGradient	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
Prod	Tidx={int32,int64} T={bfloat16,complex64,float,int32,int64,uint32,uint64}
Qr	T={float}
QuantizeAndDequantizeV2	T={bfloat16,float}
QuantizeAndDequantizeV3	T={bfloat16,float}
RFFT
RFFT2D
RFFT3D
RGBToHSV	T={bfloat16,float}
RandomShuffle	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
RandomStandardNormal	T={int32,int64} dtype={bfloat16,float}
RandomUniform	T={int32,int64} dtype={bfloat16,float}
RandomUniformInt	T={int32,int64} Tout={int32,int64}
Range	Tidx={bfloat16,float,int32,int64}
Rank	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
ReadVariableOp	dtype={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
Real	Tout={float} T={complex64}
RealDiv	T={bfloat16,complex64,float,int32,int64}
Reciprocal	T={bfloat16,complex64,float,int32,int64}
ReciprocalGrad	T={bfloat16,complex64,float}
RecvTPUEmbeddingActivations
Relu	T={bfloat16,float,int32,int64,uint32,uint64}
Relu6	T={bfloat16,float,int32,int64,uint32,uint64}
Relu6Grad	T={bfloat16,float,int32,int64,uint32,uint64}
ReluGrad	T={bfloat16,float,int32,int64,uint32,uint64}
Reshape	Tshape={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
ResizeBilinear	T={bfloat16,float,int32,int64}
ResizeBilinearGrad	T={bfloat16,float}
ResizeNearestNeighbor	T={float,int32,int64}
ResourceApplyAdaMax	T={bfloat16,float}
ResourceApplyAdadelta	T={bfloat16,float}
ResourceApplyAdagrad	T={bfloat16,float}
ResourceApplyAdagradDA	T={bfloat16,float}
ResourceApplyAdam	T={bfloat16,float}
ResourceApplyAddSign	T={bfloat16,float}
ResourceApplyCenteredRMSProp	T={bfloat16,float}
ResourceApplyFtrl	T={bfloat16,float}
ResourceApplyFtrlV2	T={bfloat16,float}
ResourceApplyGradientDescent	T={bfloat16,float}
ResourceApplyKerasMomentum	T={bfloat16,float}
ResourceApplyMomentum	T={bfloat16,float}
ResourceApplyPowerSign	T={bfloat16,float}
ResourceApplyProximalAdagrad	T={bfloat16,float}
ResourceApplyProximalGradientDescent	T={bfloat16,float}
ResourceApplyRMSProp	T={bfloat16,float}
ResourceGather	Tindices={int32,int64} dtype={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
ResourceScatterAdd	Tindices={int32,int64} dtype={bfloat16,complex64,float,int32,int64,uint32,uint64}
ResourceScatterDiv	Tindices={int32,int64} dtype={bfloat16,complex64,float,int32,int64,uint32,uint64}
ResourceScatterMax	Tindices={int32,int64} dtype={bfloat16,complex64,float,int32,int64,uint32,uint64}
ResourceScatterMin	Tindices={int32,int64} dtype={bfloat16,complex64,float,int32,int64,uint32,uint64}
ResourceScatterMul	Tindices={int32,int64} dtype={bfloat16,complex64,float,int32,int64,uint32,uint64}
ResourceScatterNdAdd	Tindices={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
ResourceScatterNdSub	Tindices={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
ResourceScatterNdUpdate	Tindices={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
ResourceScatterSub	Tindices={int32,int64} dtype={bfloat16,complex64,float,int32,int64,uint32,uint64}
ResourceScatterUpdate	Tindices={int32,int64} dtype={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
ResourceStridedSliceAssign	Index={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
Reverse	T={bool,complex64,float,int32,int64}
ReverseSequence	Tlen={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
ReverseV2	T={bfloat16,bool,complex64,float,int32,int64} Tidx={int32,int64}
RightShift	T={int32,int64,uint32,uint64}
Rint	T={bfloat16,float}
Round	T={bfloat16,complex64,float,int32,int64}
Rsqrt	T={bfloat16,complex64,float}
RsqrtGrad	T={bfloat16,complex64,float}
ScatterNd	Tindices={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
Select	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
Selu	T={bfloat16,float}
SeluGrad	T={bfloat16,float}
SendTPUEmbeddingGradients
Shape	out_type={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
ShapeN	out_type={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
Sigmoid	T={bfloat16,complex64,float}
SigmoidGrad	T={bfloat16,complex64,float}
Sign	T={bfloat16,complex64,float,int32,int64}
Sin	T={bfloat16,complex64,float}
Sinh	T={bfloat16,complex64,float}
Size	out_type={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
Slice	Index={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
Snapshot	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
Softmax	T={bfloat16,float}
SoftmaxCrossEntropyWithLogits	T={bfloat16,float}
Softplus	T={bfloat16,float}
SoftplusGrad	T={bfloat16,float}
Softsign	T={bfloat16,float}
SoftsignGrad	T={bfloat16,float}
SpaceToBatch	Tpaddings={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
SpaceToBatchND	Tblock_shape={int32,int64} Tpaddings={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
SpaceToDepth	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
SparseMatMul	Tb={bfloat16,float} Ta={bfloat16,float}
SparseSoftmaxCrossEntropyWithLogits	Tlabels={int32,int64} T={bfloat16,float}
SparseToDense	Tindices={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
Split	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
SplitV	Tlen={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
Sqrt	T={bfloat16,complex64,float}
SqrtGrad	T={bfloat16,complex64,float}
Square	T={bfloat16,complex64,float,int32,int64}
SquaredDifference	T={bfloat16,complex64,float,int32,int64}
Squeeze	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
StackCloseV2
StackPopV2	elem_type={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
StackPushV2	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
StackV2	elem_type={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
StatelessIf	Tout={bfloat16,bool,complex64,float,int32,int64,resource,uint32,uint64} Tin={bfloat16,bool,complex64,float,int32,int64,resource,uint32,uint64} Tcond={bfloat16,bool,complex64,float,int32,int64,resource,uint32,uint64}
StatelessMultinomial	output_dtype={int32,int64} Tseed={int32} T={bfloat16,float}
StatelessRandomNormal	Tseed={int32} T={int32,int64} dtype={bfloat16,float}
StatelessRandomUniform	Tseed={int32} T={int32,int64} dtype={bfloat16,float}
StatelessRandomUniformInt	Tseed={int32} T={int32,int64} dtype={int32,int64}
StatelessTruncatedNormal	Tseed={int32} T={int32,int64} dtype={bfloat16,float}
StatelessWhile	T={bfloat16,bool,complex64,float,int32,int64,resource,uint32,uint64}
StopGradient	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
StridedSlice	Index={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
StridedSliceGrad	Index={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
Sub	T={bfloat16,complex64,float,int32,int64}
Sum	Tidx={int32,int64} T={bfloat16,complex64,float,int32,int64,uint32,uint64}
SymbolicGradient	Tout={bfloat16,bool,complex64,float,int32,int64,uint32,uint64} Tin={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
TPUEmbeddingActivations
Tan	T={bfloat16,complex64,float,int32,int64}
Tanh	T={bfloat16,complex64,float}
TanhGrad	T={bfloat16,complex64,float}
TensorArrayCloseV3
TensorArrayConcatV3	dtype={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
TensorArrayGatherV3	dtype={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
TensorArrayGradV3
TensorArrayReadV3	dtype={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
TensorArrayScatterV3	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
TensorArraySizeV3
TensorArraySplitV3	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
TensorArrayV3	dtype={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
TensorArrayWriteV3	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
TensorListElementShape	shape_type={int32,int64}
TensorListPopBack	element_dtype={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
TensorListPushBack	element_dtype={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
TensorListReserve	shape_type={int32,int64} element_dtype={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
Tile	Tmultiples={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
TopKV2	T={bfloat16,float,int32,uint32}
Transpose	Tperm={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
TruncateDiv	T={bfloat16,complex64,float,int32,int64}
TruncateMod	T={bfloat16,float,int32,int64}
TruncatedNormal	T={int32,int64} dtype={float}
Unpack	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
UnsortedSegmentMax	Tnumsegments={int32,int64} Tindices={int32,int64} T={bfloat16,float,int32,int64,uint32,uint64}
UnsortedSegmentMin	Tnumsegments={int32,int64} Tindices={int32,int64} T={bfloat16,float,int32,int64,uint32,uint64}
UnsortedSegmentProd	Tnumsegments={int32,int64} Tindices={int32,int64} T={bfloat16,complex64,float,int32,int64,uint32,uint64}
UnsortedSegmentSum	Tnumsegments={int32,int64} Tindices={int32,int64} T={bfloat16,complex64,float,int32,int64,uint32,uint64}
VarIsInitializedOp
VariableShape	out_type={int32,int64}
While	T={bfloat16,bool,complex64,float,int32,int64,resource,uint32,uint64}
Xdivy	T={complex64,float}
XlaBroadcastHelper	Tindices={int32,int64} T={bfloat16,complex64,float,int32,int64,uint32,uint64}
XlaConv	Tindices={int32,int64} T={bfloat16,complex64,float,int32,int64,uint32,uint64}
XlaDequantize
XlaDot	T={bfloat16,complex64,float,int32,int64,uint32,uint64}
XlaDynamicSlice	Tindices={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
XlaDynamicUpdateSlice	Tindices={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
XlaHostCompute	Toutputs={bfloat16,bool,complex64,float,int32,int64,uint32,uint64} Tinputs={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
XlaIf	Tout={bfloat16,bool,complex64,float,int32,int64,resource,uint32,uint64} Tin={bfloat16,bool,complex64,float,int32,int64,resource,uint32,uint64} Tcond={bfloat16,bool,complex64,float,int32,int64,resource,uint32,uint64}
XlaKeyValueSort	V={bfloat16,bool,complex64,float,int32,int64,uint32,uint64} K={bfloat16,float,int32,int64,uint32,uint64}
XlaPad	Tindices={int32,int64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
XlaRecv	dtype={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
XlaRecvFromHost	Toutput={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
XlaReduce	T={bfloat16,complex64,float,int32,int64,uint32,uint64}
XlaReduceWindow	Tindices={int32,int64} T={bfloat16,complex64,float,int32,int64,uint32,uint64}
XlaSelectAndScatter	Tindices={int32,int64} T={bfloat16,complex64,float,int32,int64,uint32,uint64}
XlaSend	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
XlaSendToHost	Tinput={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
XlaSort	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
XlaWhile	T={bfloat16,bool,complex64,float,int32,int64,resource,uint32,uint64}
Xlogy	T={complex64,float}
ZerosLike	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
_Arg	T={bfloat16,bool,complex64,float,int32,int64,resource,uint32,uint64}
_ArrayToList	out_types={bfloat16,bool,complex64,float,int32,int64,uint32,uint64} T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
_ListToArray	T={bfloat16,bool,complex64,float,int32,int64,uint32,uint64} Tin={bfloat16,bool,complex64,float,int32,int64,uint32,uint64}
_Retval	T={bfloat16,bool,complex64,float,int32,int64,resource,uint32,uint64}