Home
last modified time | relevance | path

Searched refs:b_idx (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/
Dgraph_topology_view_test.cc74 const absl::optional<int> b_idx = graph_view.GetNodeIndex(*b_by_name); in TEST_F() local
75 ASSERT_TRUE(b_idx.has_value()); in TEST_F()
76 EXPECT_EQ(b_idx.value(), 1); in TEST_F()
/external/tensorflow/tensorflow/core/util/sparse/
Ddim_comparator.h73 const TTypes<int64>::ConstMatrix& b_idx, in cmp() argument
77 const int64 b = b_idx(b_row, d); in cmp()
/external/tensorflow/tensorflow/core/kernels/
Dsparse_conditional_accumulator.h410 inline int cmp(std::vector<int64>* a_idx, const Tensor* b_idx, in cmp() argument
413 const int64 b = b_idx->vec<int64>()(b_row); in cmp()