Home
last modified time | relevance | path

Searched defs:dim_val (Results 1 – 5 of 5) 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.cc2223 auto dim_val = c->Value(dim); in SliceHelper() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc2506 int64_t dim_val = (*dim_attr.begin()).getSExtValue(); in InferExpandDimsOpType() local
Dtf_ops_n_z.cc1602 int64_t dim_val = dim.value().getSExtValue(); in verify() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc3517 int64_t dim_val = dim.value().getSExtValue(); in matchAndRewrite() local