Searched refs:b_map (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | dense_to_sparse.cc | 53 std::vector<int>* b_map, std::vector<int>* b_size) { in PopulateEncodingParams() argument
64 *b_map = {}; in PopulateEncodingParams()
72 b_map->push_back(i); in PopulateEncodingParams()
114 std::vector<int> b_map = {}; in CalculateBlockSparsity() local
115 PopulateEncodingParams(block_size, &traversal_order, &format, &b_map, in CalculateBlockSparsity()
120 shape, traversal_order, format, b_size, b_map); in CalculateBlockSparsity()
130 shape, traversal_order, format, b_size, b_map); in CalculateBlockSparsity()
141 shape, traversal_order, format, b_size, b_map); in CalculateBlockSparsity()
240 std::vector<int> b_map = {}; in BuildSparsityParameterAttribute() local
241 PopulateEncodingParams(block_size, &traversal_order, &format, &b_map, in BuildSparsityParameterAttribute()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | call_graph_test.cc | 449 HloInstruction* b_map = b_computation->root_instruction(); in TEST_F() local
488 b_map, c_computation->root_instruction()), in TEST_F()
491 EXPECT_EQ(call_graph->NearestAncestorsInSameComputation(b_map, entry_while), in TEST_F()
493 EXPECT_EQ(call_graph->NearestAncestorsInSameComputation(b_map, a_call), in TEST_F()
497 EXPECT_EQ(call_graph->NearestAncestorsInSameComputation(a_while, b_map), in TEST_F()
|
| /external/rust/crates/itertools/tests/ |
| D | test_std.rs | 894 let b_map = b.into_iter().map(|f| f as i32); in diff_mismatch() localVariable
895 let diff = it::diff_with(a.iter(), b_map, |a, b| *a == b); in diff_mismatch()
908 let b_map = b.into_iter().map(|f| f as i32); in diff_longer() localVariable
909 let diff = it::diff_with(a.iter(), b_map, |a, b| *a == b); in diff_longer()
922 let b_map = b.into_iter().map(|f| f as i32); in diff_shorter() localVariable
923 let diff = it::diff_with(a.iter(), b_map, |a, b| *a == b); in diff_shorter()
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | stats_ops.cc | 1030 auto& b_map = f_insert_result.first->second; in Compute() local
1032 b_map.insert(BucketMapIterator::value_type( in Compute()
|