Searched refs:split_index (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | batch_resource_base.h | 78 int split_index = 0; member
101 int split_index, AsyncOpKernel::DoneCallback done_callback);
|
| D | batch_resource_base.cc | 212 int split_index, AsyncOpKernel::DoneCallback done_callback) { in CreateSplitTask() argument
221 task->split_index = split_index; in CreateSplitTask()
313 batch_components->split_index = 0; in RegisterInput()
646 std::vector<Tensor>& tensor_vector = (*task.output)[task.split_index]; in SplitOutputTensors()
|
| /external/openscreen/osp/demo/ |
| D | osp_demo.cc | 339 size_t split_index = line.find_first_of(' '); in SeparateCommandFromArguments() local
342 std::string command = line.substr(0, split_index); in SeparateCommandFromArguments()
344 split_index < line.size() ? line.substr(split_index + 1) : std::string(); in SeparateCommandFromArguments()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_tensor_from_variant_op.cc | 173 int split_index = dims + i; in NestedStackRaggedTensors() local
183 output_ragged->mutable_splits(split_index)->vec<SPLIT_TYPE>(); in NestedStackRaggedTensors()
|
| /external/libaom/av1/encoder/ |
| D | var_based_part.c | 1060 const int split_index = 21 + i2 + j; in fill_variance_tree_leaves() local
1062 force_split[split_index] = PART_EVAL_ALL; in fill_variance_tree_leaves()
1086 force_split[split_index] = PART_EVAL_ONLY_SPLIT; in fill_variance_tree_leaves()
1107 force_split[split_index] = PART_EVAL_ONLY_SPLIT; in fill_variance_tree_leaves()
1115 force_split[split_index] = PART_EVAL_ALL; in fill_variance_tree_leaves()
1474 const int split_index = 21 + i2 + j; in av1_choose_var_based_partitioning() local
1485 force_split[split_index] = in av1_choose_var_based_partitioning()
1596 const int split_index = 21 + i2 + j; in av1_choose_var_based_partitioning() local
1608 force_split[split_index])) { in av1_choose_var_based_partitioning()
|
| /external/rust/crates/itertools/src/ |
| D | lib.rs | 3739 let mut split_index = 0; in partition() localVariable
3753 split_index += 1; in partition()
3755 split_index in partition()
|
| /external/rust/crates/itertools/tests/ |
| D | quick.rs | 862 let split_index = itertools::partition(&mut ap, |x| *x >= 0); localVariable
863 let parted = (0..split_index).all(|i| ap[i] >= 0) &&
864 (split_index..a.len()).all(|i| ap[i] < 0);
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 1343 for (int split_index = 0; split_index < num_splits; split_index++) { in TransformSplitVIntoSupportedOps() local
1344 int split_size = size_splits[split_index] == -1 in TransformSplitVIntoSupportedOps()
1346 : size_splits[split_index]; in TransformSplitVIntoSupportedOps()
1373 int lite_output_index = node->outputs->data[split_index]; in TransformSplitVIntoSupportedOps()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 681 const int split_index, Graph* graph) { in CreatePadNode() argument
708 absl::StrCat(split_node->name(), "/pad_shard_", split_index))); in CreatePadNode()
713 pad_def.add_input(absl::StrCat(split_node->name(), ":", split_index)); in CreatePadNode()
718 graph->AddEdge(split_node, split_index, pad_node, 0); in CreatePadNode()
|
| /external/libvpx/vp9/encoder/ |
| D | vp9_encodeframe.c | 1562 const int split_index = 5 + i2 + j; in choose_partitioning() local
1564 force_split[split_index] = 0; in choose_partitioning()
1582 force_split[split_index] = 1; in choose_partitioning()
1601 force_split[split_index] = 1; in choose_partitioning()
1610 force_split[split_index] = 0; in choose_partitioning()
|