Searched refs:sparse_shapes (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | example_parsing_ops.cc | 149 OpOutputList sparse_shapes; in Compute() local
153 OP_REQUIRES_OK(ctx, ctx->output_list("sparse_shapes", &sparse_shapes)); in Compute()
160 sparse_shapes.set(d, result.sparse_shapes[d]); in Compute()
245 OpOutputList sparse_shapes; in Compute() local
249 OP_REQUIRES_OK(ctx, ctx->output_list("sparse_shapes", &sparse_shapes)); in Compute()
256 sparse_shapes.set(d, result.sparse_shapes[d]); in Compute()
410 context_sparse_shapes.set(d, context_result.sparse_shapes[d]); in Compute()
415 feature_list_sparse_shapes.set(d, feature_list_result.sparse_shapes[d]); in Compute()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.h | 89 std::vector<Tensor> sparse_shapes; member
|
| D | example_proto_fast_parsing.cc | 1106 result->sparse_shapes.emplace_back(DT_INT64, TensorShape({2})); in FastParseExample()
1107 auto shapes_shape_t = result->sparse_shapes.back().vec<int64>(); in FastParseExample()
1286 result->sparse_shapes.emplace_back(DT_INT64, TensorShape({1})); in FastParseSingleExample()
1495 Tensor* out_dense_shape = &result->sparse_shapes[d]; in FastParseSingleExample()
1568 result->sparse_shapes[d].vec<int64>()(0) = 0; in FastParseSingleExample()
1993 context_result->sparse_shapes.resize(context_config.sparse.size()); in FastParseSequenceExample()
1997 feature_list_result->sparse_shapes.resize(feature_list_config.sparse.size()); in FastParseSequenceExample()
2127 context_result->sparse_shapes[t] = Tensor(DT_INT64, TensorShape({2})); in FastParseSequenceExample()
2148 auto out_shape = context_result->sparse_shapes[t].vec<int64>(); in FastParseSequenceExample()
2325 feature_list_result->sparse_shapes[t] = Tensor(DT_INT64, TensorShape({3})); in FastParseSequenceExample()
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | parsing_ops.py | 869 (sparse_indices, sparse_values, sparse_shapes, dense_values) = outputs
873 in zip(sparse_indices, sparse_values, sparse_shapes)]
2122 (sparse_indices, sparse_values, sparse_shapes, dense_values) = outputs
2127 shape) in zip(sparse_indices, sparse_values, sparse_shapes)
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | parse_example_dataset_op.cc | 320 serialized_sparse_t(2) = example_result.sparse_shapes[d]; in MapFunc()
|
| /external/tensorflow/tensorflow/cc/saved_model/testdata/half_plus_two_pbtxt/00000123/ |
| D | saved_model.pbtxt | 224 name: "sparse_shapes"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 11029 …s []tf.Shape) (sparse_indices []tf.Output, sparse_values []tf.Output, sparse_shapes []tf.Output, d…
11055 if sparse_shapes, idx, err = makeOutputList(op, idx, "sparse_shapes"); err != nil {
11063 return sparse_indices, sparse_values, sparse_shapes, dense_values
16838 …s []tf.Shape) (sparse_indices []tf.Output, sparse_values []tf.Output, sparse_shapes []tf.Output, d…
16864 if sparse_shapes, idx, err = makeOutputList(op, idx, "sparse_shapes"); err != nil {
16872 return sparse_indices, sparse_values, sparse_shapes, dense_values
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 22247 name: "sparse_shapes"
22510 name: "sparse_shapes"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 19597 name: "sparse_shapes"
|
| D | ops_history.v2.pbtxt | 43455 name: "sparse_shapes"
43718 name: "sparse_shapes"
|
| D | ops_history.v1.pbtxt | 43978 name: "sparse_shapes"
44241 name: "sparse_shapes"
|