Home
last modified time | relevance | path

Searched defs:max_rank (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dcwise_ops.cc127 int max_rank = std::max(lhs_shape.dims(), rhs_shape.dims()); in Compile() local
/external/tensorflow/tensorflow/core/kernels/boosted_trees/quantiles/
Dweighted_quantiles_summary.h80 WeightType max_rank; member
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Drank_specialization.cc598 const SmallVector<Value, 8> &shapes, Value max_rank, in RecusivelyMaterializeTargetRankSpecializationCases() argument
646 Value max_rank; in MaterializeGenericRankSpecializationCases() local
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dbufferize.cc114 Value max_rank; in matchAndRewrite() local
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dquantile_ops.cc324 float max_rank = summary_values(i, 3); in Compute() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc203 int max_rank = 1; in CwiseOutputElementCount() local
Dgraph_properties.cc2370 const int64_t max_rank = 4; in PropagateShapes() local
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc91 int64_t max_rank = -1; in VerifyOperandsHaveSameShapesOrBroadcastableShape() local
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc444 std::vector<int64> OpTest::RandomDims(int min_rank, int max_rank, in RandomDims()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc312 const int max_rank = nvinfer1::Dims::MAX_DIMS + (use_implicit_batch ? 1 : 0); in ValidateTensorProperties() local