Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/boosted_trees/proto/
Dquantiles.proto23 float max_rank = 4; field
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dcwise_ops.cc69 int max_rank = std::max(lhs_shape.dims(), rhs_shape.dims()); in Compile() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_fusible.cc49 int64 max_rank = -1; in LayoutsAreReduceInputFusionFriendly() local
/external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/
Dcwise_op_vectorizer.cc69 Output max_rank = ops::Maximum(scope, rank_0, rank_1); in ExpandDimsForBroadcast() local
/external/tensorflow/tensorflow/core/kernels/boosted_trees/quantiles/
Dweighted_quantiles_summary.h79 WeightType max_rank; member
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/quantiles/
Dweighted_quantiles_summary.h71 WeightType max_rank; member
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dquantile_ops.cc254 float max_rank = summary_values(i, 3); in Compute() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1905 const int64 max_rank = 4; in PropagateShapes() local
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc443 std::vector<int64> OpTest::RandomDims(int min_rank, int max_rank, in RandomDims()