Searched refs:batch_prefix (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | numa_map_and_batch_dataset_op.cc | 627 string batch_prefix = strings::StrCat(prefix, "_batch_", i); in Save() local
629 strings::StrCat(batch_prefix, "_code"), in Save()
633 writer->WriteScalar(strings::StrCat(batch_prefix, "_msg"), in Save()
636 strings::StrCat(batch_prefix, "_error_index"), in Save()
641 strings::StrCat(batch_prefix, "_output_size"), in Save()
645 strings::StrCat(batch_prefix, "_output_", j); in Save()
676 string batch_prefix = strings::StrCat(prefix, "_batch_", i); in Restore() local
677 if (!reader->Contains(strings::StrCat(batch_prefix, "_code"))) { in Restore()
684 strings::StrCat(batch_prefix, "_code"), &code_int)); in Restore()
689 strings::StrCat(batch_prefix, "_msg"), &error_message)); in Restore()
[all …]
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | unroll_batch_matmul.cc | 124 const std::vector<int>& batch_prefix) { in UnrollBatchMatMulRecursion() argument
127 const int current_dim_size = dims_vec[batch_prefix.size()]; in UnrollBatchMatMulRecursion()
130 if (batch_prefix.size() + 3 == dims_vec.size()) { in UnrollBatchMatMulRecursion()
133 std::vector<int> new_batch_prefix = batch_prefix; in UnrollBatchMatMulRecursion()
141 std::vector<int> new_batch_prefix = batch_prefix; in UnrollBatchMatMulRecursion()
|