Home
last modified time | relevance | path

Searched refs:split_index (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dbatch_resource_base.h76 int split_index = 0; member
99 int split_index, AsyncOpKernel::DoneCallback done_callback);
Dbatch_resource_base.cc132 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/
Dvar_based_part.c786 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/
Dosp_demo.cc337 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/
Dragged_tensor_from_variant_op.cc116 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/
Dlib.rs3151 let mut split_index = 0; in partition() localVariable
3165 split_index += 1; in partition()
3167 split_index in partition()
/external/rust/crates/itertools/tests/
Dquick.rs866 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/
Ddistributed_tpu_rewrite_pass.cc665 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/
Dvp9_encodeframe.c1594 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()