Home
last modified time | relevance | path

Searched refs:output_ragged (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_from_variant_op.cc83 const int output_ragged_rank, RaggedTensorVariant* output_ragged) { in NestedStackRaggedTensors() argument
84 output_ragged->mutable_nested_splits()->reserve(output_ragged_rank); in NestedStackRaggedTensors()
90 output_ragged->append_splits(Tensor(DataTypeToEnum<SPLIT_TYPE>::value, in NestedStackRaggedTensors()
92 auto splits_vec = output_ragged->mutable_splits(i)->vec<SPLIT_TYPE>(); in NestedStackRaggedTensors()
101 output_ragged->append_splits( in NestedStackRaggedTensors()
104 output_ragged->mutable_splits(dims - 1)->vec<SPLIT_TYPE>(); in NestedStackRaggedTensors()
123 output_ragged->append_splits( in NestedStackRaggedTensors()
126 output_ragged->mutable_splits(split_index)->vec<SPLIT_TYPE>(); in NestedStackRaggedTensors()
172 output_ragged->set_values( in NestedStackRaggedTensors()
175 output_ragged->mutable_values()->flat_outer_dims<VALUE_TYPE, 2>(); in NestedStackRaggedTensors()
[all …]
/external/tensorflow/tensorflow/python/keras/layers/
Dpooling_test.py70 output_ragged = model.predict(ragged_data, steps=1)
78 self.assertAllEqual(output_ragged, output_dense)
90 output_ragged = model.predict(ragged_data, steps=1)
97 self.assertAllEqual(output_ragged, output_dense)
108 output_ragged = model.predict(ragged_data, steps=1)
112 self.assertAllEqual(output_ragged, expected_output)
Dwrappers_test.py427 output_ragged = model_1.predict(ragged_data, steps=1)
437 output_ragged = convert_ragged_tensor_value(output_ragged)
438 self.assertAllEqual(output_ragged.to_tensor(), output_dense)
456 output_ragged = model_1.predict(ragged_data, steps=1)
465 output_ragged = convert_ragged_tensor_value(output_ragged)
466 self.assertAllEqual(output_ragged.to_tensor(), output_dense)
Drecurrent_test.py1602 output_ragged = model.predict(ragged_data, steps=1)
1611 self.assertAllClose(output_dense, output_ragged)
1620 output_ragged = model.predict(ragged_data, steps=1)
1629 self.assertAllClose(output_dense, output_ragged)
1673 output_ragged = model.predict(ragged_data, steps=1)
1674 self.assertAllClose(output_ragged.ragged_rank, ragged_data.ragged_rank)
1675 self.assertAllClose(output_ragged.row_splits, ragged_data.row_splits)
1686 self.assertAllClose(output_ragged, output_dense)
/external/tensorflow/tensorflow/core/util/
Dexample_proto_fast_parsing.cc578 std::vector<SparseBuffer>* output_ragged, in FastParseSerializedExample() argument
582 DCHECK(output_ragged != nullptr); in FastParseSerializedExample()
804 SparseBuffer& out = is_ragged ? (*output_ragged)[d] : (*output_sparse)[d]; in FastParseSerializedExample()
923 SparseBuffer& out = (*output_ragged)[d]; in FastParseSerializedExample()