Searched refs:compute_data_format (Results 1 – 6 of 6) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
H A D | conv_grad_input_ops.cc | 224 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 D | conv_ops_3d.cc | 355 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 D | conv_ops.cc | 901 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 D | conv_grad_ops_3d.cc | 1379 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 D | conv_grad_filter_ops.cc | 843 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 D | conv_ops_fused_impl.h | 479 const TensorFormat compute_data_format = 591 compute_data_format, // compute_data_format
|