Home
last modified time | relevance | path

Searched defs:data_format_ (Results 1 – 21 of 21) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dmaxpooling_op.cc351 TensorFormat data_format_; member in tensorflow::MaxPoolingGradOp
452 TensorFormat data_format_; member in tensorflow::MaxPoolingGradOp
679 TensorFormat data_format_; member in tensorflow::MaxPoolingGradGradOp
791 TensorFormat data_format_; member in tensorflow::MaxPoolingGradGradOp
859 TensorFormat data_format_; member in tensorflow::MaxPoolingNoMaskOp
942 TensorFormat data_format_; member in tensorflow::MaxPoolingNoMaskV2Op
1155 TensorFormat data_format_; member in tensorflow::MaxPoolingGradWithArgmaxOp
1328 TensorFormat data_format_; member in tensorflow::MaxPoolingNoMaskOp
1425 TensorFormat data_format_; member in tensorflow::MaxPoolingNoMaskV2Op
H A Davgpooling_op.cc119 TensorFormat data_format_; member in tensorflow::AvgPoolingOp
224 TensorFormat data_format_; member in tensorflow::AvgPoolingOp
406 TensorFormat data_format_; member in tensorflow::AvgPoolingGradOp
486 TensorFormat data_format_; member in tensorflow::AvgPoolingGradOp
632 TensorFormat data_format_; member in tensorflow::AvgPoolingGradOpCustomGPUKernel
H A Dbias_op.cc145 TensorFormat data_format_; member in tensorflow::BiasOp
224 TensorFormat data_format_; member in tensorflow::BiasGradOp
282 TensorFormat data_format_; member in tensorflow::BiasOp
376 TensorFormat data_format_; member in tensorflow::BiasAddParams
532 TensorFormat data_format_; member in tensorflow::BiasGradOp
H A Dconv_grad_ops_3d.cc292 TensorFormat data_format_; member in tensorflow::Conv3DBackpropInputOp
640 TensorFormat data_format_; member in tensorflow::Conv3DCustomBackpropInputOp
808 TensorFormat data_format_; member in tensorflow::Conv3DBackpropFilterOp
1117 TensorFormat data_format_; member in tensorflow::Conv3DCustomBackpropFilterOp
1579 TensorFormat data_format_; member in tensorflow::Conv3DBackpropInputOp
1951 TensorFormat data_format_; member in tensorflow::Conv3DBackpropFilterOp
H A Dpooling_ops_3d.cc204 TensorFormat data_format_; member in tensorflow::Pooling3DOp
387 TensorFormat data_format_; member in tensorflow::MaxPooling3dGradOp
557 TensorFormat data_format_; member in tensorflow::AvgPooling3dGradOp
765 TensorFormat data_format_; member in tensorflow::MaxPooling3dGradGradOp
H A Dconv_grad_filter_ops.cc376 TensorFormat data_format_; member in tensorflow::Conv2DBackpropFilterOp
622 TensorFormat data_format_; member in tensorflow::Conv2DCustomBackpropFilterOp
H A Ddepthtospace_op.cc142 TensorFormat data_format_; member in tensorflow::DepthToSpaceOp
H A Dspacetodepth_op.cc160 TensorFormat data_format_; member in tensorflow::SpaceToDepthOp
H A Ddepthwise_conv_grad_op.cc700 TensorFormat data_format_; member in tensorflow::DepthwiseConv2dNativeBackpropInputOp
1240 TensorFormat data_format_; member in tensorflow::DepthwiseConv2dNativeBackpropFilterOp
H A Dpooling_ops_common.h299 TensorFormat data_format_; variable
H A Ddepthwise_conv_op.cc499 TensorFormat data_format_; member in tensorflow::DepthwiseConv2dNativeOp
H A Dconv_ops_3d.cc202 TensorFormat data_format_; member in tensorflow::Conv3DOp
H A Dconv_ops_using_gemm.cc560 TensorFormat data_format_; member in tensorflow::Conv2DUsingGemmOp
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dpooling_ops.cc138 TensorFormat data_format_ = FORMAT_NHWC; member in tensorflow::__anona9add1350111::PoolingOp
418 TensorFormat data_format_ = FORMAT_NHWC; member in tensorflow::__anona9add1350111::MaxPoolGradOp
520 TensorFormat data_format_ = FORMAT_NHWC; member in tensorflow::__anona9add1350111::AvgPoolGradOp
697 TensorFormat data_format_ = FORMAT_NHWC; member in tensorflow::__anona9add1350111::MaxPoolGradGradOp
H A Dbatch_norm_op.cc201 TensorFormat data_format_; member in tensorflow::__anona7fdd7270111::FusedBatchNormOp
348 TensorFormat data_format_; member in tensorflow::__anona7fdd7270111::FusedBatchNormGradOp
H A Dbias_ops.cc73 TensorFormat data_format_; member in tensorflow::__anon64c9cebc0111::BiasOp
H A Ddepthtospace_op.cc181 TensorFormat data_format_; member in tensorflow::__anon44cd0dce0111::DepthToSpaceOp
H A Dspacetodepth_op.cc185 TensorFormat data_format_; member in tensorflow::__anonb48f31ee0111::SpaceToDepthOp
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_conv_ops.h60 TensorFormat data_format_; variable
666 TensorFormat data_format_; // NCHW or NHWC variable
H A Dmkl_concat_op.cc454 TensorFormat data_format_; member in tensorflow::MklConcatOp
H A Dmkl_conv_ops.cc1181 TensorFormat data_format_; member in tensorflow::MklConvOp