| /external/tensorflow/tensorflow/core/kernels/ |
| D | example_parsing_ops.cc | 260 OpOutputList sparse_shapes; in WriteOutput() local
264 TF_RETURN_IF_ERROR(ctx->output_list("sparse_shapes", &sparse_shapes)); in WriteOutput()
271 sparse_shapes.set(d, result.sparse_shapes[d]); in WriteOutput()
372 OpOutputList sparse_shapes; in Compute() local
376 OP_REQUIRES_OK(ctx, ctx->output_list("sparse_shapes", &sparse_shapes)); in Compute()
383 sparse_shapes.set(d, result.sparse_shapes[d]); in Compute()
729 context_sparse_shapes.set(d, context_result.sparse_shapes[d]); in WriteOutput()
734 feature_list_sparse_shapes.set(d, feature_list_result.sparse_shapes[d]); in WriteOutput()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | ParseSingleExample.pbtxt | 21 name: "sparse_shapes"
|
| D | ParseExample.pbtxt | 35 name: "sparse_shapes"
|
| D | ParseExampleV2.pbtxt | 37 name: "sparse_shapes"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | ParseSingleExample.pbtxt | 21 name: "sparse_shapes"
|
| D | ParseExample.pbtxt | 35 name: "sparse_shapes"
|
| D | ParseExampleV2.pbtxt | 37 name: "sparse_shapes"
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.h | 121 std::vector<Tensor> sparse_shapes; member
|
| D | example_proto_fast_parsing.cc | 1241 result->sparse_shapes.reserve(config.sparse.size()); in FastParseExample()
1269 result->sparse_shapes.emplace_back(DT_INT64, TensorShape({2})); in FastParseExample()
1270 auto shapes_shape_t = result->sparse_shapes.back().vec<int64>(); in FastParseExample()
1474 result->sparse_shapes.reserve(config.sparse.size()); in FastParseSingleExample()
1497 result->sparse_shapes.emplace_back(DT_INT64, TensorShape({1})); in FastParseSingleExample()
1721 Tensor* out_dense_shape = &result->sparse_shapes[d]; in FastParseSingleExample()
1800 result->sparse_shapes[d].vec<int64>()(0) = 0; in FastParseSingleExample()
2359 context_result->sparse_shapes[t] = in ParseContextSparseFeatures()
2364 auto out_shape = context_result->sparse_shapes[t].vec<int64>(); in ParseContextSparseFeatures()
2650 sequence_result->sparse_shapes[t] = in ParseSequenceSparseFeatures()
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | parsing_ops.py | 363 (sparse_indices, sparse_values, sparse_shapes, dense_values,
371 in zip(sparse_indices, sparse_values, sparse_shapes)]
|
| /external/tensorflow/tensorflow/lite/kernels/parse_example/ |
| D | parse_example.cc | 67 std::vector<TfLiteTensor*> sparse_shapes; member
491 TfLiteTensor* dense_shape = result->sparse_shapes[d]; in FastParseExampleLite()
799 data->got.sparse_shapes.push_back(parse_output); in PrepareParseExample()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | parse_example_dataset_op.cc | 675 serialized_sparse_t(2) = example_result.sparse_shapes[d]; in ParseExample()
|
| /external/tensorflow/tensorflow/cc/saved_model/testdata/half_plus_two_pbtxt/00000123/ |
| D | saved_model.pbtxt | 224 name: "sparse_shapes"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops.td | 489 Variadic<TF_Int64Tensor>:$sparse_shapes, // len(sparse_types)
524 Variadic<TF_Int64Tensor>:$sparse_shapes, // len(sparse_types)
|
| D | tf_ops_n_z.cc | 380 if (op.sparse_shapes().size() != sparse_types_count) { in Verify()
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 28080 …s []tf.Shape) (sparse_indices []tf.Output, sparse_values []tf.Output, sparse_shapes []tf.Output, d…
28106 if sparse_shapes, idx, err = makeOutputList(op, idx, "sparse_shapes"); err != nil {
28122 return sparse_indices, sparse_values, sparse_shapes, dense_values, ragged_values, ragged_row_splits
40322 …s []tf.Shape) (sparse_indices []tf.Output, sparse_values []tf.Output, sparse_shapes []tf.Output, d…
40348 if sparse_shapes, idx, err = makeOutputList(op, idx, "sparse_shapes"); err != nil {
40356 return sparse_indices, sparse_values, sparse_shapes, dense_values
45733 …s []tf.Shape) (sparse_indices []tf.Output, sparse_values []tf.Output, sparse_shapes []tf.Output, d…
45759 if sparse_shapes, idx, err = makeOutputList(op, idx, "sparse_shapes"); err != nil {
45767 return sparse_indices, sparse_values, sparse_shapes, dense_values
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | tf-ops.mlir | 3568 …// expected-error @+1 {{output 'sparse_shapes' should have same length as attribute 'sparse_types'…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 28274 name: "sparse_shapes"
28592 name: "sparse_shapes"
29153 name: "sparse_shapes"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 19597 name: "sparse_shapes"
|
| D | ops_history.v2.pbtxt | 42792 name: "sparse_shapes"
43055 name: "sparse_shapes"
|