/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...] |