| /external/tensorflow/tensorflow/contrib/libsvm/ops/ |
| libsvm_ops.cc | 36 c->set_output(1, c->Matrix(InferenceContext::kUnknownDim,
37 InferenceContext::kUnknownDim));
38 c->set_output(2, c->Vector(InferenceContext::kUnknownDim));
39 c->set_output(3, c->Vector(InferenceContext::kUnknownDim));
|
| /external/tensorflow/tensorflow/core/ops/ |
| tpu_ordinal_selector_ops.cc | 26 c->Vector(shape_inference::InferenceContext::kUnknownDim));
|
| sparse_ops.cc | 36 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim,
37 InferenceContext::kUnknownDim));
38 c->set_output(1, c->Vector(InferenceContext::kUnknownDim));
79 0, c->Matrix(InferenceContext::kUnknownDim, c->Dim(a_shape, 0)));
80 c->set_output(1, c->Vector(InferenceContext::kUnknownDim));
148 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, 3));
163 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim,
164 InferenceContext::kUnknownDim));
165 c->set_output(1, c->Vector(InferenceContext::kUnknownDim));
166 c->set_output(2, c->Vector(InferenceContext::kUnknownDim));
[all...] |
| ragged_math_ops.cc | 66 int64 rt_nested_splits_dim = InferenceContext::kUnknownDim;
|
| sdca_ops.cc | 38 {c->Matrix(InferenceContext::kUnknownDim, c->MakeDim(4))});
|
| ctc_ops.cc | 120 c->set_output(out_idx++, c->Matrix(InferenceContext::kUnknownDim, 2));
123 c->set_output(out_idx++, c->Vector(InferenceContext::kUnknownDim));
|
| cudnn_rnn_ops.cc | 331 c->set_output(i, c->Matrix(InferenceContext::kUnknownDim,
332 InferenceContext::kUnknownDim));
336 c->set_output(num_params + i, c->Vector(InferenceContext::kUnknownDim));
357 c->set_output(0, c->Vector(InferenceContext::kUnknownDim));
|
| string_ops.cc | 185 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, 2));
186 c->set_output(1, c->Vector(InferenceContext::kUnknownDim));
203 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, 2));
204 c->set_output(1, c->Vector(InferenceContext::kUnknownDim));
|
| candidate_sampling_ops.cc | 159 ShapeHandle v = c->Vector(InferenceContext::kUnknownDim);
|
| io_ops.cc | 382 ShapeHandle out = c->Vector(InferenceContext::kUnknownDim);
399 ShapeHandle out = c->Vector(InferenceContext::kUnknownDim);
490 c->set_output(0, c->Vector(InferenceContext::kUnknownDim));
|
| parsing_ops.cc | 38 c->input(0), c->Vector(InferenceContext::kUnknownDim), &out));
313 c->Concatenate(c->Vector(InferenceContext::kUnknownDim), s, &s));
|
| image_ops.cc | 85 c->set_output(0, c->MakeShape({InferenceContext::kUnknownDim,
86 InferenceContext::kUnknownDim, channels_dim}));
545 c->set_output(0, c->MakeShape({InferenceContext::kUnknownDim,
546 InferenceContext::kUnknownDim,
547 InferenceContext::kUnknownDim, 3}));
[all...] |
| math_ops.cc | 996 c->Concatenate(c->Vector(InferenceContext::kUnknownDim), subshape, &out));
[all...] |
| array_ops.cc | [all...] |
| data_flow_ops.cc | 341 n_shape = c->Vector(InferenceContext::kUnknownDim);
367 return DequeueManyV2Shape(c, c->Vector(InferenceContext::kUnknownDim));
637 prepend_rank != InferenceContext::kUnknownDim) {
[all...] |
| resource_variable_ops.cc | 243 : InferenceContext::kUnknownDim;
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/ops/ |
| prediction_ops.cc | 36 c->set_output(0, {c->Matrix(InferenceContext::kUnknownDim,
52 c->set_output(0, {c->Matrix(InferenceContext::kUnknownDim,
56 c->set_output(2, {c->Matrix(InferenceContext::kUnknownDim,
57 InferenceContext::kUnknownDim)});
179 {c->Vector(InferenceContext::kUnknownDim)});
|
| /external/tensorflow/tensorflow/contrib/cloud/ops/ |
| bigquery_reader_ops.cc | 67 c->set_output(0, c->Vector(InferenceContext::kUnknownDim));
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/core/ops/ |
| unpack_path_op.cc | 44 int64 num_nodes = InferenceContext::kUnknownDim;
|
| /external/tensorflow/tensorflow/core/framework/ |
| shape_inference.cc | 30 constexpr int64 InferenceContext::kUnknownDim;
318 if (dim_val == kUnknownDim) {
428 if (existing == kUnknownDim) {
538 if (v0 == kUnknownDim || v1 == kUnknownDim || v0 != v1) {
588 if (v0 == kUnknownDim) {
589 if (v1 != kUnknownDim) {
592 } else if (v1 == kUnknownDim) {
[all...] |
| shape_inference.h | 129 // val must be non-negative or InferenceContext::kUnknownDim.
160 static constexpr int64 kUnknownDim = -1;
368 return Value(d) != kUnknownDim;
504 inline DimensionHandle UnknownDim() { return MakeDim(kUnknownDim); }
794 inline Dimension::Dimension() : value_(InferenceContext::kUnknownDim) {}
796 DCHECK(value >= 0 || value == InferenceContext::kUnknownDim)
798 "InferenceContext::kUnknownDim but got "
812 DCHECK(val >= 0 || val == InferenceContext::kUnknownDim)
814 "InferenceContext::kUnknownDim but got "
[all...] |
| shape_inference_testutil_test.cc | 90 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, 2));
|
| shape_inference_test.cc | 118 EXPECT_EQ(InferenceContext::kUnknownDim,
119 c.Value(InferenceContext::kUnknownDim));
280 EXPECT_EQ(InferenceContext::kUnknownDim, c.Value(d));
534 {S({2, InferenceContext::kUnknownDim, 2, 1,
535 InferenceContext::kUnknownDim})},
550 EXPECT_EQ(InferenceContext::kUnknownDim, c.Value(out));
553 EXPECT_EQ(InferenceContext::kUnknownDim, c.Value(out));
557 EXPECT_EQ(InferenceContext::kUnknownDim, c.Value(out));
573 EXPECT_EQ(InferenceContext::kUnknownDim, c.Value(out));
575 EXPECT_EQ(InferenceContext::kUnknownDim, c.Value(out))
[all...] |
| shape_inference_testutil.cc | 245 dims.push_back(manager->MakeDim(InferenceContext::kUnknownDim));
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| shape_refiner_test.cc | 888 0, c->MakeShape({1, shape_inference::InferenceContext::kUnknownDim, 3,
889 shape_inference::InferenceContext::kUnknownDim, 5}));
899 c->MakeShape({6, shape_inference::InferenceContext::kUnknownDim, 8}));
[all...] |
