Home
last modified time | relevance | path

Searched defs:dim_val (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dshape_inference.cc267 int64_t dim_val = Value(Dim(s, i)); in NumElements() local
Dcommon_shape_fns.cc2172 auto dim_val = c->Value(dim); in SliceHelper() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc1542 int64_t dim_val = dim.value().getSExtValue(); in Verify() local
Dtf_ops_a_m.cc2338 int64_t dim_val = (*dim_attr.begin()).getSExtValue(); in InferExpandDimsOpType() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_linalg.cc2775 Value dim_val = b.create<SelectOp>(loc, use_missing_dim_val, in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc3709 int64_t dim_val = dim.value().getSExtValue(); in matchAndRewrite() local