Home
last modified time | relevance | path

Searched defs:data_format_str (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dcommon_shape_fns.cc587 string data_format_str, filter_format_str; in Conv2DShapeImpl() local
886 string data_format_str; in Conv2DBackpropInputShape() local
1007 string data_format_str; in DepthwiseConv2DNativeShapeImpl() local
1110 string data_format_str; in AvgPoolShape() local
1187 string data_format_str; in FusedBatchNormShape() local
1237 string data_format_str; in FusedBatchNormExShape() local
1260 string data_format_str; in FusedBatchNormGradShape() local
1309 string data_format_str; in FusedBatchNormGradExShape() local
1587 string data_format_str; in MaxPoolShapeImpl() local
1694 string data_format_str; in MaxPoolV2Shape() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dbatch_norm_op.cc46 string data_format_str; in FusedBatchNormOp() local
247 string data_format_str; in FusedBatchNormGradOp() local
H A Dpooling_ops.cc165 std::string data_format_str; in MaxPoolOp() local
259 string data_format_str; in AvgPoolOp() local
H A Ddepthtospace_op.cc29 string data_format_str; in DepthToSpaceOp() local
H A Dspacetodepth_op.cc29 string data_format_str; in SpaceToDepthOp() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_tfconv_op.h65 string data_format_str, DataType op_data_type, in ConvertMklToTf()
H A Dmkl_input_conversion_op.cc277 string data_format_str; member in tensorflow::MklInputConversionOp
H A Dmkl_conv_ops.h619 string data_format_str; in MklConvBackpropCommonOp() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Ddepthtospace_op.cc47 string data_format_str; in DepthToSpaceOp() local
H A Dspacetodepth_op.cc62 string data_format_str; in SpaceToDepthOp() local
H A Dmaxpooling_op.cc1087 string data_format_str; in MaxPoolingGradWithArgmaxOp() local
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Darray_ops.cc2508 string data_format_str; in __anon38bbb0e83702() local
2562 string data_format_str; in __anon38bbb0e83802() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dmkl_layout_pass.cc1520 string data_format_str; in NonDepthBatchWisePoolRewrite() local