Home
last modified time | relevance | path

Searched defs:spatial_dim_index (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddynamic_padder.cc973 for (int64_t spatial_dim_index = 0; in RewriteDynamicConvolutionInputGrad() local
1043 for (int64_t spatial_dim_index = 0; in RewriteDynamicConvolutionForward() local
1104 for (int64_t spatial_dim_index = 0; in RewriteDynamicConvolutionKernelGrad() local
H A Ddynamic_dimension_inference.cc752 for (int64_t spatial_dim_index = 0; in HandleDynamicConvolutionForward() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dcommon_shape_fns.cc532 for (int spatial_dim_index = 0, end = spatial_dims.size(); in DimensionsFromShape() local
561 for (int spatial_dim_index = 0, end = spatial_dims.size(); in ShapeFromDimensions() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_ops_a_m.cc410 llvm::StringRef crop_b_name) -> LogicalResult { in verify()