Home
last modified time | relevance | path

Searched defs:splits (Results 1 – 25 of 53) sorted by relevance

123

/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_variant.h58 const Tensor& splits(int i) const { return nested_splits_[i]; } in splits() function
63 void append_splits(const Tensor& splits) { nested_splits_.push_back(splits); } in append_splits()
Dragged_gather_op.cc169 const auto& splits = params_nested_splits[dim]; in MakeSplits() local
193 const auto& splits = params_nested_splits[dim]; in ValidateSplits() local
223 Tensor* splits; in WriteSplits() local
Dragged_tensor_variant.cc37 for (const auto& splits : nested_splits_) { in Encode() local
Dragged_tensor_to_variant_op_test.cc58 for (const auto& splits : ragged_splits) { in BuildEncodeRaggedTensorGraph() local
73 Tensor splits(DataTypeToEnum<SPLIT_TYPE>::v(), in CreateVariantFromRagged() local
Dragged_tensor_to_sparse_kernel_test.cc52 for (const auto& splits : rt_nested_splits) { in BuildRaggedTensorToSparseGraph() local
Dragged_gather_op_test.cc54 for (const auto& splits : params_nested_splits) { in BuildRaggedGatherGraph() local
Dragged_tensor_to_sparse_kernel.cc139 const auto& splits = rt_nested_splits[dim]; in Compute() local
/external/tensorflow/tensorflow/core/ops/
Dragged_conversion_ops.cc171 ShapeHandle splits = c->input(i); in RaggedTensorToSparseShapeFn() local
196 shape_inference::ShapeHandle splits = c->input(i); in RaggedTensorToVariantShapeFn() local
Dragged_array_ops.cc135 ShapeHandle splits = c->input(i); in RaggedGatherShapeFn() local
/external/tensorflow/tensorflow/python/ops/ragged/
Dsegment_id_ops.py36 def row_splits_to_segment_ids(splits, name=None, out_type=None): argument
Dragged_util.py70 def repeat_ranges(params, splits, repeats): argument
Dragged_concat_ops.py293 def _copy_row_shape(rt_inputs, splits): argument
/external/tensorflow/tensorflow/core/framework/
Drun_handler_util.cc38 std::vector<string> splits = str_util::Split(val, ","); in ParamFromEnvWithDefault() local
59 std::vector<string> splits = str_util::Split(val, ","); in ParamFromEnvWithDefault() local
Dextension_type_variant_test.cc44 const std::vector<int64>& splits) { in Make2DRaggedTensor()
/external/opencensus-java/contrib/zpages/src/main/java/io/opencensus/contrib/zpages/
DZPageHttpHandler.java79 List<String> splits = Splitter.on("=").splitToList(param); in uriQueryToMap() local
/external/tensorflow/tensorflow/python/ops/
Draw_ops_test.py68 def testStringNGramsBadDataSplits(self, splits): argument
/external/tensorflow/tensorflow/core/util/
Dsaved_tensor_slice_util.cc95 std::vector<string> splits = str_util::Split(shape_and_slice, ' '); in ParseShapeAndSlice() local
/external/perfetto/src/traced/probes/system_info/
Dsystem_info_data_source.cc87 auto splits = base::SplitString(line, ":"); in Start() local
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/sentencepiece/
Dsentencepiece_tokenizer_op.cc81 std::vector<int32> splits; in Compute() local
Dsentencepiece_tokenizer_tflite.cc86 std::vector<int32> splits; in Eval() local
/external/tensorflow/tensorflow/compiler/mlir/xla/
Doperator_writer_gen.cc51 std::pair<StringRef, StringRef> splits = attr.getReturnType().rsplit("::"); in GetDefaultAttrExport() local
/external/rust/crates/textwrap/src/
Dsplitting.rs107 let mut splits = Vec::new(); in split_points() localVariable
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_util.cc35 const std::vector<std::string> splits = absl::StrSplit(key, '|'); in ParseCompilationCacheKey() local
/external/Reactive-Extensions/RxCpp/Rx/v2/examples/linesfrombytes/
Dmain.cpp78 vector<string> splits(cursor, end); in main() local
/external/rust/crates/cexpr/tests/
Dclang.rs65 let mut splits = value.split(|c| *c == b'U'); in test_definition() localVariable

123