Home
last modified time | relevance | path

Searched refs:kUnknownDim (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/core/ops/
Dsparse_ops.cc36 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, in SparseSparseMinOrMaxShapeFn()
37 InferenceContext::kUnknownDim)); in SparseSparseMinOrMaxShapeFn()
38 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in SparseSparseMinOrMaxShapeFn()
79 0, c->Matrix(InferenceContext::kUnknownDim, c->Dim(a_shape, 0))); in __anona7ffb82a0302()
80 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in __anona7ffb82a0302()
148 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, 3)); in __anona7ffb82a0602()
163 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, in __anona7ffb82a0702()
164 InferenceContext::kUnknownDim)); in __anona7ffb82a0702()
165 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in __anona7ffb82a0702()
166 c->set_output(2, c->Vector(InferenceContext::kUnknownDim)); in __anona7ffb82a0702()
[all …]
Dstring_ops.cc185 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, 2)); in __anonb4ab91930502()
186 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in __anonb4ab91930502()
203 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, 2)); in __anonb4ab91930602()
204 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in __anonb4ab91930602()
Dcudnn_rnn_ops.cc331 c->set_output(i, c->Matrix(InferenceContext::kUnknownDim, in __anone1d60d010902()
332 InferenceContext::kUnknownDim)); in __anone1d60d010902()
336 c->set_output(num_params + i, c->Vector(InferenceContext::kUnknownDim)); in __anone1d60d010902()
357 c->set_output(0, c->Vector(InferenceContext::kUnknownDim)); in __anone1d60d010a02()
Dctc_ops.cc120 c->set_output(out_idx++, c->Matrix(InferenceContext::kUnknownDim, 2)); in __anon78c05cd60302()
123 c->set_output(out_idx++, c->Vector(InferenceContext::kUnknownDim)); in __anon78c05cd60302()
Dtpu_ordinal_selector_ops.cc26 c->Vector(shape_inference::InferenceContext::kUnknownDim)); in __anon5f9864dd0102()
Dio_ops.cc382 ShapeHandle out = c->Vector(InferenceContext::kUnknownDim); in __anone43acdb40902()
399 ShapeHandle out = c->Vector(InferenceContext::kUnknownDim); in __anone43acdb40a02()
490 c->set_output(0, c->Vector(InferenceContext::kUnknownDim)); in __anone43acdb40e02()
Dimage_ops.cc85 c->set_output(0, c->MakeShape({InferenceContext::kUnknownDim, in DecodeImageShapeFn()
86 InferenceContext::kUnknownDim, channels_dim})); in DecodeImageShapeFn()
545 c->set_output(0, c->MakeShape({InferenceContext::kUnknownDim, in __anon6a71d27f0e02()
546 InferenceContext::kUnknownDim, in __anon6a71d27f0e02()
547 InferenceContext::kUnknownDim, 3})); in __anon6a71d27f0e02()
Dragged_math_ops.cc66 int64 rt_nested_splits_dim = InferenceContext::kUnknownDim; in RaggedRangeShapeFn()
Dsdca_ops.cc38 {c->Matrix(InferenceContext::kUnknownDim, c->MakeDim(4))}); in ApplySdcaOptimizerShapeFn()
Dparsing_ops.cc38 c->input(0), c->Vector(InferenceContext::kUnknownDim), &out)); in __anon952516500102()
313 c->Concatenate(c->Vector(InferenceContext::kUnknownDim), s, &s)); in __anon952516500502()
Dmath_ops.cc996 c->Concatenate(c->Vector(InferenceContext::kUnknownDim), subshape, &out)); in SegmentReductionShapeFn()
1020 c->Concatenate(c->Vector(InferenceContext::kUnknownDim), subshape, &out)); in SparseSegmentReductionShapeFn()
1047 dim0_shape = c->Vector(InferenceContext::kUnknownDim); in SparseSegmentReductionGradShapeFn()
1088 TF_RETURN_IF_ERROR(c->Concatenate(c->Vector(InferenceContext::kUnknownDim), in SparseSegmentReductionWithNumSegmentsShapeFn()
1362 c->set_output(0, c->Vector(InferenceContext::kUnknownDim)); in __anonb22bfa860902()
1394 c->set_output(0, c->Vector(InferenceContext::kUnknownDim)); in __anonb22bfa860a02()
Dcandidate_sampling_ops.cc159 ShapeHandle v = c->Vector(InferenceContext::kUnknownDim); in __anon78463f730202()
Darray_ops.cc1355 c->set_output(0, c->Vector(InferenceContext::kUnknownDim)); in __anon7c94107b1e02()
1371 c->set_output(0, c->Vector(InferenceContext::kUnknownDim)); in __anon7c94107b1f02()
1385 auto uniq = c->Vector(InferenceContext::kUnknownDim); in __anon7c94107b2002()
1402 auto uniq = c->Vector(InferenceContext::kUnknownDim); in __anon7c94107b2102()
1746 c->set_output(0, c->Vector(InferenceContext::kUnknownDim)); in __anon7c94107b2902()
1770 c->set_output(0, c->Vector(InferenceContext::kUnknownDim)); in __anon7c94107b2a02()
1771 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in __anon7c94107b2a02()
2050 ShapeHandle out = c->Vector(InferenceContext::kUnknownDim); in __anon7c94107b3102()
2477 c->MakeShape({batch_size_dim, InferenceContext::kUnknownDim, in __anon7c94107b3902()
2478 InferenceContext::kUnknownDim, output_depth_dim}); in __anon7c94107b3902()
[all …]
Ddata_flow_ops.cc341 n_shape = c->Vector(InferenceContext::kUnknownDim); in __anon4a7ac08d0602()
367 return DequeueManyV2Shape(c, c->Vector(InferenceContext::kUnknownDim)); in __anon4a7ac08d0702()
637 prepend_rank != InferenceContext::kUnknownDim) { in __anon4a7ac08d1302()
Dresource_variable_ops.cc243 : InferenceContext::kUnknownDim; in VariableShapeShapeFn()
/external/tensorflow/tensorflow/contrib/libsvm/ops/
Dlibsvm_ops.cc36 c->set_output(1, c->Matrix(InferenceContext::kUnknownDim, in __anonbd00c5ed0102()
37 InferenceContext::kUnknownDim)); in __anonbd00c5ed0102()
38 c->set_output(2, c->Vector(InferenceContext::kUnknownDim)); in __anonbd00c5ed0102()
39 c->set_output(3, c->Vector(InferenceContext::kUnknownDim)); in __anonbd00c5ed0102()
/external/tensorflow/tensorflow/contrib/boosted_trees/ops/
Dprediction_ops.cc36 c->set_output(0, {c->Matrix(InferenceContext::kUnknownDim, in ApplyGradientTreesPredictionShapeFn()
52 c->set_output(0, {c->Matrix(InferenceContext::kUnknownDim, in ApplyGradientTreesPredictionVerboseShapeFn()
56 c->set_output(2, {c->Matrix(InferenceContext::kUnknownDim, in ApplyGradientTreesPredictionVerboseShapeFn()
57 InferenceContext::kUnknownDim)}); in ApplyGradientTreesPredictionVerboseShapeFn()
179 {c->Vector(InferenceContext::kUnknownDim)}); in __anonbe66d2360102()
/external/tensorflow/tensorflow/core/framework/
Dshape_inference.cc30 constexpr int64 InferenceContext::kUnknownDim; member in tensorflow::shape_inference::InferenceContext
318 if (dim_val == kUnknownDim) { in NumElements()
428 if (existing == kUnknownDim) { in WithValue()
538 if (v0 == kUnknownDim || v1 == kUnknownDim || v0 != v1) { in Relax()
588 if (v0 == kUnknownDim) { in Merge()
589 if (v1 != kUnknownDim) { in Merge()
592 } else if (v1 == kUnknownDim) { in Merge()
1046 } else if (first_value == kUnknownDim || second_value == kUnknownDim) { in Add()
1071 } else if (first_value == kUnknownDim || second_value == kUnknownDim) { in Subtract()
1100 } else if (first_value == kUnknownDim || second_value == kUnknownDim) { in Multiply()
[all …]
Dshape_inference.h160 static constexpr int64 kUnknownDim = -1;
368 return Value(d) != kUnknownDim; in ValueKnown()
504 inline DimensionHandle UnknownDim() { return MakeDim(kUnknownDim); } in UnknownDim()
794 inline Dimension::Dimension() : value_(InferenceContext::kUnknownDim) {} in Dimension()
796 DCHECK(value >= 0 || value == InferenceContext::kUnknownDim) in Dimension()
812 DCHECK(val >= 0 || val == InferenceContext::kUnknownDim) in DimensionOrConstant()
Dshape_inference_test.cc118 EXPECT_EQ(InferenceContext::kUnknownDim, in TEST_F()
119 c.Value(InferenceContext::kUnknownDim)); in TEST_F()
280 EXPECT_EQ(InferenceContext::kUnknownDim, c.Value(d)); in TEST_F()
534 {S({2, InferenceContext::kUnknownDim, 2, 1, in TEST_F()
535 InferenceContext::kUnknownDim})}, in TEST_F()
550 EXPECT_EQ(InferenceContext::kUnknownDim, c.Value(out)); in TEST_F()
553 EXPECT_EQ(InferenceContext::kUnknownDim, c.Value(out)); in TEST_F()
557 EXPECT_EQ(InferenceContext::kUnknownDim, c.Value(out)); in TEST_F()
573 EXPECT_EQ(InferenceContext::kUnknownDim, c.Value(out)); in TEST_F()
575 EXPECT_EQ(InferenceContext::kUnknownDim, c.Value(out)); in TEST_F()
[all …]
Dshape_inference_testutil_test.cc90 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, 2)); in TEST()
Dshape_inference_testutil.cc245 dims.push_back(manager->MakeDim(InferenceContext::kUnknownDim)); in MakeShapeFromString()
/external/tensorflow/tensorflow/contrib/cloud/ops/
Dbigquery_reader_ops.cc67 c->set_output(0, c->Vector(InferenceContext::kUnknownDim)); in __anon97f137840202()
/external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/core/ops/
Dunpack_path_op.cc44 int64 num_nodes = InferenceContext::kUnknownDim; in __anona9c96fc40102()
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner_test.cc888 0, c->MakeShape({1, shape_inference::InferenceContext::kUnknownDim, 3, in __anon7afc939f0c02()
889 shape_inference::InferenceContext::kUnknownDim, 5})); in __anon7afc939f0c02()
899 c->MakeShape({6, shape_inference::InferenceContext::kUnknownDim, 8})); in __anon7afc939f0d02()
1377 shp = ctx->MakeShape({shape_inference::InferenceContext::kUnknownDim, 7}); in TEST_F()

12