Searched refs:fewer_dims_optimization (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | bcast.h | 82 const bool fewer_dims_optimization = true,
132 const bool fewer_dims_optimization, in BCastList() argument
155 if (all_equal && TF_PREDICT_TRUE(fewer_dims_optimization)) { in BCastList()
230 if (!TF_PREDICT_TRUE(fewer_dims_optimization)) { in BCastList()
256 } else if (TF_PREDICT_TRUE(fewer_dims_optimization) && in BCastList()
364 BCast(const Vec& x, const Vec& y, const bool fewer_dims_optimization = true,
366 : BCastList<2>({x, y}, fewer_dims_optimization,
|
| D | bcast_test.cc | 27 const bool fewer_dims_optimization = true) { in BCast() argument
28 tensorflow::BCast b(x, y, fewer_dims_optimization); in BCast()
46 const bool fewer_dims_optimization = true) { in BCastBatchIndices() argument
47 tensorflow::BCast b(x, y, fewer_dims_optimization, in BCastBatchIndices()
58 const bool fewer_dims_optimization = true) { in BCastList3() argument
59 tensorflow::BCastList<3> b({x, y, z}, fewer_dims_optimization); in BCastList3()
|