Searched defs:num_concats (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| 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 …]
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing_test.cc | 351 auto num_concats = 1 + rng->Rand32() % 4; in Fuzz() local
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 1305 const std::vector<int64> num_concats( in CreateXlaConcatNode() local
|