Searched defs:num_concats (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | sharding_util_ops_test.cc | 597 const std::vector<int32> num_concats = {1, 1}; in TEST() local
645 const std::vector<int32> num_concats = {1, 1}; in TEST() local
681 const std::vector<int32> num_concats = {1, 1}; in TEST() local
716 const std::vector<int32> num_concats = {1, 1}; in TEST() local
763 const std::vector<int32> num_concats = {1, 1}; in TEST() local
791 absl::Span<const int32> num_concats, in CreateConcatTensorGraph()
822 absl::Span<const int32> num_concats, absl::Span<const int32> paddings, in CreateConcatResourceGraph()
893 const std::vector<int32> num_concats = {1, 1, 0}; in TEST_P() local
925 const std::vector<int32> num_concats = {2}; in TEST_P() local
941 const std::vector<int32> num_concats = {1, 1}; in TEST_P() local
[all …]
|
| H A D | sharding_util_ops.cc | 738 MaybeUnpadAndAssignState(absl::Span<const int32> num_concats, in MaybeUnpadAndAssignState()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | example_proto_fast_parsing_test.cc | 355 auto num_concats = 1 + rng->Rand32() % 4; in Fuzz() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | distributed_tpu_rewrite_pass.cc | 1295 const std::vector<int64_t> num_concats( in CreateXlaConcatNode() local
|