Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Ddata_format_ops.cc116 const int full_dim_count = src_format_.size(); in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
H A Ddata_format_vec_permute.cc83 const int full_dim_count = src_format->size(); in Validate() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Dprepare_tf.cc623 int full_dim_count = padded_begin.size(); in RewriteEllipsisMask() local
659 int full_dim_count = padding_val.size(); in PadStridedSliceAttributeArray() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Ddata_format_ops.cc148 const int full_dim_count = src_format_.size(); in Compute() local