Searched refs:output_splits (Results 1 – 4 of 4) sorted by relevance
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/sentencepiece/ |
| D | sentencepiece_tokenizer_tflite.cc | 62 TfLiteTensor& output_splits = in Prepare() local
64 SetTensorToDynamic(&output_splits); in Prepare()
106 TfLiteTensor& output_splits = in Eval() local
110 context, &output_splits, in Eval()
112 int32_t* output_splits_flat = output_splits.data.i32; in Eval()
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_math_ops.py | 239 output_splits = array_ops.concat([
247 data_row_to_out_row_start = array_ops.gather(output_splits, segment_ids)
259 output_splits[-1], separator)
261 output_values, output_splits, validate=False)
|
| D | ragged_string_ops.py | 809 output, output_splits = gen_string_ops.string_n_grams(
821 values=output, row_splits=output_splits, validate=False)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | xla_sharding_util.cc | 468 const auto output_splits = dimension_and_output_splits.value(); in ValidateAndGetTiledExecuteOutputShape() local
478 if (output_shape_at_dim % output_splits != 0) { in ValidateAndGetTiledExecuteOutputShape()
485 dimension_index, output_splits, output_shape_at_dim)); in ValidateAndGetTiledExecuteOutputShape()
489 output_shape[dimension_index] / output_splits; in ValidateAndGetTiledExecuteOutputShape()
|