Home
last modified time | relevance | path

Searched defs:arg_dims (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc297 auto arg_dims = llvm::drop_begin(arg_ty.getShape(), 1); in fold() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier.cc4619 std::vector<int64> arg_dims = arg->dimensions(); in HandleReduce() local