Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/tfrt/run_handler_thread_pool/
Drun_handler_util.cc40 std::vector<std::string> splits = tensorflow::str_util::Split(val, ","); in ParamFromEnvWithDefault() local
61 std::vector<std::string> splits = tensorflow::str_util::Split(val, ","); in ParamFromEnvWithDefault() local
/external/ComputeLibrary/tests/validation/fixtures/
DSplitFixture.h51 void setup(TensorShape shape, unsigned int axis, unsigned int splits, DataType data_type) in setup()
64 …rType> compute_target(const TensorShape &shape, unsigned int axis, unsigned int splits, DataType d… in compute_target()
109 …T>> compute_reference(const TensorShape &shape, unsigned int axis, unsigned int splits, DataType d… in compute_reference()
/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.cc175 const auto& splits = params_nested_splits[dim]; in MakeSplits() local
199 const auto& splits = params_nested_splits[dim]; in ValidateSplits() local
229 Tensor* splits; in WriteSplits() local
Dragged_tensor_variant.cc37 for (const auto& splits : nested_splits_) { in Encode() local
Dragged_tensor_to_variant_op_test.h61 for (const auto& splits : ragged_splits) { in BuildEncodeRaggedTensorGraph() local
Dcomposite_tensor_variant_test.cc47 const std::vector<int64_t>& splits) { in Make2DRaggedTensor()
/external/tensorflow/tensorflow/core/ops/
Dragged_conversion_ops.cc172 ShapeHandle splits = c->input(i); in RaggedTensorToSparseShapeFn() local
197 shape_inference::ShapeHandle splits = c->input(i); in RaggedTensorToVariantShapeFn() local
Dragged_array_ops.cc149 ShapeHandle splits = c->input(i); in RaggedGatherShapeFn() local
/external/tensorflow/tensorflow/python/ops/ragged/
Dsegment_id_ops.py32 def row_splits_to_segment_ids(splits, name=None, out_type=None): argument
Dragged_util.py66 def repeat_ranges(params, splits, repeats): argument
Dragged_concat_ops.py295 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
/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.py63 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.cc88 auto splits = base::SplitString(line, ":"); in Start() local
/external/tensorflow/tensorflow/compiler/mlir/xla/
Doperator_writer_gen.cc51 std::pair<StringRef, StringRef> splits = attr.getReturnType().rsplit("::"); in GetDefaultAttrExport() 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/lite/delegates/hexagon/builders/
Dconv_2d_builder.h39 int splits = 0; member
/external/ComputeLibrary/tests/datasets/
DSplitDataset.h91 void add_config(TensorShape shape, unsigned int axis, unsigned int splits) in add_config()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_util.cc35 const std::vector<std::string> splits = absl::StrSplit(key, '|'); in ParseCompilationCacheKey() local
/external/rust/crates/textwrap/src/
Dword_splitters.rs147 let mut splits = Vec::new(); in split_points() localVariable
/external/zstd/lib/compress/
Dzstd_ldm.c98 size_t* splits, unsigned* numSplits) in ZSTD_ldm_gear_feed()
276 size_t* const splits = ldmState->splitIndices; in ZSTD_ldm_fillHashTable() local
348 size_t* const splits = ldmState->splitIndices; in ZSTD_ldm_generateSequences_internal() local

123