Searched refs:input_exprs (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/tools/ |
| D | saved_model_cli.py | 659 input_exprs = preprocess_input_exprs_arg_string(input_exprs_str)
694 for input_tensor_key, py_expr_evaluated in input_exprs.items():
743 if not args.inputs and not args.input_exprs and not args.input_examples:
748 args.inputs, args.input_exprs, args.input_examples)
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 701 SmallVector<AffineExpr, 2> input_exprs; in getIndexingMaps() local
702 input_exprs.resize(result_type.getRank()); in getIndexingMaps()
704 input_exprs[permutation.value().getZExtValue()] = in getIndexingMaps()
708 AffineMap::get(nloops, /*symbolCount=*/0, input_exprs, b->getContext()), in getIndexingMaps()
1022 SmallVector<AffineExpr, 2> input_exprs; in getIndexingMaps() local
1023 input_exprs.reserve(nloops); in getIndexingMaps()
1025 input_exprs.push_back(b->getAffineDimExpr(i)); in getIndexingMaps()
1030 input_exprs[i] = b->getAffineConstantExpr(n - 1) - input_exprs[i]; in getIndexingMaps()
1033 AffineMap::get(nloops, /*symbolCount=*/0, input_exprs, b->getContext()), in getIndexingMaps()
|