Searched refs:split_index (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/batching_util/ |
D | batch_resource_base.h | 76 int split_index = 0; member 99 int split_index, AsyncOpKernel::DoneCallback done_callback);
|
D | batch_resource_base.cc | 132 int split_index, AsyncOpKernel::DoneCallback done_callback) { in CreateSplitTask() argument 141 task->split_index = split_index; in CreateSplitTask() 216 batch_components->split_index = 0; in RegisterInput() 539 std::vector<Tensor>& tensor_vector = (*task.output)[task.split_index]; in SplitOutputTensors()
|
/external/libaom/libaom/av1/encoder/ |
D | var_based_part.c | 786 const int split_index = 21 + i2 + j; in av1_choose_var_based_partitioning() local 788 force_split[split_index] = 0; in av1_choose_var_based_partitioning() 813 force_split[split_index] = 1; in av1_choose_var_based_partitioning() 833 force_split[split_index] = 1; in av1_choose_var_based_partitioning() 841 force_split[split_index] = 0; in av1_choose_var_based_partitioning() 866 const int split_index = 21 + i2 + j; in av1_choose_var_based_partitioning() local 877 force_split[split_index] = 1; in av1_choose_var_based_partitioning() 967 const int split_index = 21 + i2 + j; in av1_choose_var_based_partitioning() local 979 force_split[split_index])) { in av1_choose_var_based_partitioning()
|
/external/openscreen/osp/demo/ |
D | osp_demo.cc | 337 size_t split_index = line.find_first_of(' '); in SeparateCommandFromArguments() local 340 std::string command = line.substr(0, split_index); in SeparateCommandFromArguments() 342 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 | 116 int split_index = dims + i; in NestedStackRaggedTensors() local 126 output_ragged->mutable_splits(split_index)->vec<SPLIT_TYPE>(); in NestedStackRaggedTensors()
|
/external/rust/crates/itertools/src/ |
D | lib.rs | 3151 let mut split_index = 0; in partition() localVariable 3165 split_index += 1; in partition() 3167 split_index in partition()
|
/external/rust/crates/itertools/tests/ |
D | quick.rs | 866 let split_index = itertools::partition(&mut ap, |x| *x >= 0); localVariable 867 let parted = (0..split_index).all(|i| ap[i] >= 0) && 868 (split_index..a.len()).all(|i| ap[i] < 0);
|
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
D | distributed_tpu_rewrite_pass.cc | 665 const int split_index, Graph* graph) { in CreatePadNode() argument 693 absl::StrCat(split_node->name(), "/pad_shard_", split_index))); in CreatePadNode() 698 pad_def.add_input(absl::StrCat(split_node->name(), ":", split_index)); in CreatePadNode() 704 graph->AddEdge(split_node, split_index, pad_node, 0); in CreatePadNode()
|
/external/libvpx/libvpx/vp9/encoder/ |
D | vp9_encodeframe.c | 1594 const int split_index = 5 + i2 + j; in choose_partitioning() local 1596 force_split[split_index] = 0; in choose_partitioning() 1614 force_split[split_index] = 1; in choose_partitioning() 1633 force_split[split_index] = 1; in choose_partitioning() 1642 force_split[split_index] = 0; in choose_partitioning()
|