Home
last modified time | relevance | path

Searched refs:compute_data_format (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconv_grad_input_ops.cc224 const TensorFormat compute_data_format = in operator ()() local
230 << " compute_data_format=" << ToString(compute_data_format); in operator ()()
243 compute_data_format == FORMAT_NHWC ? kComputeInNHWC : kComputeInNCHW; in operator ()()
299 if (compute_data_format == FORMAT_NCHW) { in operator ()()
301 } else if (compute_data_format == FORMAT_NHWC) { in operator ()()
305 ToString(compute_data_format))); in operator ()()
310 if (data_format == FORMAT_NHWC && compute_data_format == FORMAT_NCHW) { in operator ()()
313 compute_data_format, dims.batch_size, dims.spatial_dims[0].output_size, in operator ()()
335 compute_data_format, in operator ()()
362 compute_data_format, // compute_data_format in operator ()()
[all …]
H A Dconv_ops_3d.cc355 const TensorFormat compute_data_format = in launch() local
361 << " compute_data_format=" << ToString(compute_data_format); in launch()
363 if (data_format == FORMAT_NHWC && compute_data_format == FORMAT_NCHW) { in launch()
383 CHECK(data_format == compute_data_format) // Crash OK in launch()
386 << " compute_data_format=" << ToString(compute_data_format); in launch()
400 compute_data_format == FORMAT_NHWC ? kComputeInNHWC : kComputeInNCHW; in launch()
440 compute_data_format == FORMAT_NCHW ? FORMAT_OIHW : FORMAT_OHWI; in launch()
462 if (data_format != compute_data_format) { in launch()
492 compute_data_format, in launch()
523 if (data_format == FORMAT_NHWC && compute_data_format == FORMAT_NCHW) { in launch()
H A Dconv_ops.cc901 const TensorFormat compute_data_format = in operator ()() local
907 << " compute_data_format=" << ToString(compute_data_format); in operator ()()
988 if (data_format == FORMAT_NHWC && compute_data_format == FORMAT_NCHW) { in operator ()()
1007 CHECK(data_format == compute_data_format) // Crash OK in operator ()()
1010 << " compute_data_format=" << ToString(compute_data_format); in operator ()()
1028 compute_data_format == FORMAT_NHWC ? kComputeInNHWC : kComputeInNCHW; in operator ()()
1080 if (compute_data_format == FORMAT_NCHW) { in operator ()()
1082 } else if (compute_data_format == FORMAT_NHWC) { in operator ()()
1086 ToString(compute_data_format))); in operator ()()
1091 if (data_format != compute_data_format) { in operator ()()
[all …]
H A Dconv_grad_ops_3d.cc1379 const TensorFormat compute_data_format = in Compute() local
1385 << " compute_data_format=" << ToString(compute_data_format); in Compute()
1398 compute_data_format == FORMAT_NHWC ? kComputeInNHWC : kComputeInNCHW; in Compute()
1436 compute_data_format == FORMAT_NCHW ? FORMAT_OIHW : FORMAT_OHWI; in Compute()
1456 if (data_format_ == FORMAT_NHWC && compute_data_format == FORMAT_NCHW) { in Compute()
1478 ShapeFromFormat(compute_data_format, in Compute()
1507 compute_data_format, in Compute()
1546 ShapeFromFormat(compute_data_format, dims.batch_size, in Compute()
1559 compute_data_format, T{}); in Compute()
1564 if (data_format_ == FORMAT_NHWC && compute_data_format == FORMAT_NCHW) { in Compute()
[all …]
H A Dconv_grad_filter_ops.cc843 const TensorFormat compute_data_format = in operator ()() local
849 << " compute_data_format=" << ToString(compute_data_format); in operator ()()
862 compute_data_format == FORMAT_NHWC ? kComputeInNHWC : kComputeInNCHW; in operator ()()
909 if (data_format == FORMAT_NHWC && compute_data_format == FORMAT_NCHW) { in operator ()()
912 compute_data_format, dims.batch_size, dims.spatial_dims[0].output_size, in operator ()()
930 if (data_format == FORMAT_NHWC && compute_data_format == FORMAT_NCHW) { in operator ()()
933 compute_data_format, GetTensorDim(compatible_input, data_format, 'N'), in operator ()()
971 compute_data_format, // compute_data_format in operator ()()
1007 compute_data_format == FORMAT_NCHW ? FORMAT_OIHW : FORMAT_OHWI; in operator ()()
H A Dconv_ops_fused_impl.h479 const TensorFormat compute_data_format =
591 compute_data_format, // compute_data_format