Home
last modified time | relevance | path

Searched refs:append_splits (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_from_variant_op.cc90 output_ragged->append_splits(Tensor(DataTypeToEnum<SPLIT_TYPE>::value, in NestedStackRaggedTensors()
101 output_ragged->append_splits( in NestedStackRaggedTensors()
123 output_ragged->append_splits( in NestedStackRaggedTensors()
Dragged_tensor_to_variant_op.cc94 (*ragged_components)[i].append_splits( in UnbatchRaggedZerothDim()
148 batched_ragged_input.append_splits(ragged_nested_splits_in[i]); in Compute()
Dragged_tensor_variant.h63 void append_splits(const Tensor& splits) { nested_splits_.push_back(splits); } in append_splits() function
Dragged_tensor_to_variant_op_test.cc76 encoded.append_splits(splits); in CreateVariantFromRagged()
Dragged_tensor_from_variant_op_test.cc69 encoded.append_splits(splits); in CreateVariantFromRagged()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dparse_example_dataset_op.cc683 serialized_ragged.append_splits(example_result.ragged_splits[d]); in ParseExample()