Home
last modified time | relevance | path

Searched refs:kUnknownDim (Results 1 – 25 of 31) 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 __anone6e195410302()
80 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in __anone6e195410302()
148 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, 3)); in __anone6e195410602()
163 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, in __anone6e195410702()
164 InferenceContext::kUnknownDim)); in __anone6e195410702()
165 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in __anone6e195410702()
166 c->set_output(2, c->Vector(InferenceContext::kUnknownDim)); in __anone6e195410702()
[all …]
Dcudnn_rnn_ops.cc335 c->set_output(i, c->Matrix(InferenceContext::kUnknownDim, in __anon3a5366b80902()
336 InferenceContext::kUnknownDim)); in __anon3a5366b80902()
340 c->set_output(num_params + i, c->Vector(InferenceContext::kUnknownDim)); in __anon3a5366b80902()
371 c->set_output(i, c->Matrix(InferenceContext::kUnknownDim, in __anon3a5366b80a02()
372 InferenceContext::kUnknownDim)); in __anon3a5366b80a02()
377 c->Vector(InferenceContext::kUnknownDim)); in __anon3a5366b80a02()
398 c->set_output(0, c->Vector(InferenceContext::kUnknownDim)); in __anon3a5366b80b02()
420 c->set_output(0, c->Vector(InferenceContext::kUnknownDim)); in __anon3a5366b80c02()
Dstring_ops.cc195 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, 2)); in __anonf38d6eaa0502()
196 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in __anonf38d6eaa0502()
213 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, 2)); in __anonf38d6eaa0602()
214 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in __anonf38d6eaa0602()
Dcollective_ops.cc63 shape_inference::InferenceContext::kUnknownDim) { in __anon2a9b463d0102()
65 c->Vector(shape_inference::InferenceContext::kUnknownDim); in __anon2a9b463d0102()
Dtpu_ordinal_selector_ops.cc26 c->Vector(shape_inference::InferenceContext::kUnknownDim)); in __anone963de340102()
Dctc_ops.cc160 c->set_output(out_idx++, c->Matrix(InferenceContext::kUnknownDim, 2)); in __anon7dc6794d0402()
163 c->set_output(out_idx++, c->Vector(InferenceContext::kUnknownDim)); in __anon7dc6794d0402()
Dio_ops.cc382 ShapeHandle out = c->Vector(InferenceContext::kUnknownDim); in __anonec23b74b0902()
399 ShapeHandle out = c->Vector(InferenceContext::kUnknownDim); in __anonec23b74b0a02()
491 c->set_output(0, c->Vector(InferenceContext::kUnknownDim)); in __anonec23b74b0e02()
Dimage_ops.cc85 c->set_output(0, c->MakeShape({InferenceContext::kUnknownDim, in DecodeImageShapeFn()
86 InferenceContext::kUnknownDim, channels_dim})); in DecodeImageShapeFn()
118 c->MakeShape({InferenceContext::kUnknownDim, in DecodeImageV2ShapeFn()
119 InferenceContext::kUnknownDim, channels_dim})); in DecodeImageV2ShapeFn()
629 c->set_output(0, c->MakeShape({InferenceContext::kUnknownDim, in __anonc970e8b60e02()
630 InferenceContext::kUnknownDim, in __anonc970e8b60e02()
631 InferenceContext::kUnknownDim, 3})); in __anonc970e8b60e02()
Dsdca_ops.cc38 {c->Matrix(InferenceContext::kUnknownDim, c->MakeDim(4))}); in ApplySdcaOptimizerShapeFn()
Dragged_math_ops.cc67 int64 rt_nested_splits_dim = InferenceContext::kUnknownDim; in RaggedRangeShapeFn()
Dmath_ops.cc1133 c->Concatenate(c->Vector(InferenceContext::kUnknownDim), subshape, &out)); in SegmentReductionShapeFn()
1157 c->Concatenate(c->Vector(InferenceContext::kUnknownDim), subshape, &out)); in SparseSegmentReductionShapeFn()
1184 dim0_shape = c->Vector(InferenceContext::kUnknownDim); in SparseSegmentReductionGradShapeFn()
1225 TF_RETURN_IF_ERROR(c->Concatenate(c->Vector(InferenceContext::kUnknownDim), in SparseSegmentReductionWithNumSegmentsShapeFn()
1480 c->set_output(0, c->Vector(InferenceContext::kUnknownDim)); in __anon57f5a5dd0902()
1520 c->set_output(0, c->Vector(InferenceContext::kUnknownDim)); in __anon57f5a5dd0a02()
2077 int32 dim = dim_t == nullptr ? InferenceContext::kUnknownDim in __anon57f5a5dd1c02()
2081 ? InferenceContext::kUnknownDim in __anon57f5a5dd1c02()
Dcandidate_sampling_ops.cc159 ShapeHandle v = c->Vector(InferenceContext::kUnknownDim); in __anon3d65770a0202()
Dragged_array_ops.cc103 if (batch_size != InferenceContext::kUnknownDim) { in __anoncd0b9d3b0102()
Darray_ops.cc1449 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in UniqueIdxShapeFn()
1495 c->set_output(0, c->Vector(InferenceContext::kUnknownDim)); in __anondb9326b21e02()
1525 auto uniq = c->Vector(InferenceContext::kUnknownDim); in __anondb9326b22002()
1544 c->set_output(2, c->Vector(InferenceContext::kUnknownDim)); in __anondb9326b22102()
1901 c->set_output(0, c->Vector(InferenceContext::kUnknownDim)); in __anondb9326b22902()
1925 c->set_output(0, c->Vector(InferenceContext::kUnknownDim)); in __anondb9326b22a02()
1926 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in __anondb9326b22a02()
2205 ShapeHandle out = c->Vector(InferenceContext::kUnknownDim); in __anondb9326b23102()
2636 c->MakeShape({batch_size_dim, InferenceContext::kUnknownDim, in __anondb9326b23902()
2637 InferenceContext::kUnknownDim, output_depth_dim}); in __anondb9326b23902()
[all …]
Ddata_flow_ops.cc341 n_shape = c->Vector(InferenceContext::kUnknownDim); in __anona2f81a440602()
367 return DequeueManyV2Shape(c, c->Vector(InferenceContext::kUnknownDim)); in __anona2f81a440702()
692 prepend_rank != InferenceContext::kUnknownDim) { in __anona2f81a441702()
Dresource_variable_ops.cc226 : InferenceContext::kUnknownDim; in VariableShapeShapeFn()
/external/tensorflow/tensorflow/c/
Dc_api_experimental_test.cc157 static constexpr int64_t kUnknownDim = member in tensorflow::__anon639294630111::ShapeInferenceTest
158 shape_inference::InferenceContext::kUnknownDim;
179 /*expected_shape*/ make_shape({3, kUnknownDim})); in TEST_F()
184 /*input_shapes*/ {make_shape({kUnknownDim, 2}), make_shape({2, 4})}, in TEST_F()
186 /*expected_shape*/ make_shape({kUnknownDim, 4})); in TEST_F()
192 /*expected_shape*/ make_shape({kUnknownDim, kUnknownDim})); in TEST_F()
/external/tensorflow/tensorflow/core/framework/
Dshape_inference.cc30 constexpr int64 InferenceContext::kUnknownDim; member in tensorflow::shape_inference::InferenceContext
261 if (dim_val == kUnknownDim) { in NumElements()
370 if (existing == kUnknownDim) { in WithValue()
480 if (v0 == kUnknownDim || v1 == kUnknownDim || v0 != v1) { in Relax()
530 if (v0 == kUnknownDim) { in Merge()
531 if (v1 != kUnknownDim) { in Merge()
534 } else if (v1 == kUnknownDim) { in Merge()
1019 } else if (first_value == kUnknownDim || second_value == kUnknownDim) { in Add()
1044 } else if (first_value == kUnknownDim || second_value == kUnknownDim) { in Subtract()
1073 } else if (first_value == kUnknownDim || second_value == kUnknownDim) { in Multiply()
[all …]
Dshape_inference.h158 static constexpr int64 kUnknownDim = -1;
354 return Value(d) != kUnknownDim; in ValueKnown()
490 inline DimensionHandle UnknownDim() { return MakeDim(kUnknownDim); } in UnknownDim()
779 inline Dimension::Dimension() : value_(InferenceContext::kUnknownDim) {} in Dimension()
781 DCHECK(value >= 0 || value == InferenceContext::kUnknownDim) in Dimension()
797 DCHECK(val >= 0 || val == InferenceContext::kUnknownDim) in DimensionOrConstant()
Dshape_inference_test.cc126 EXPECT_EQ(InferenceContext::kUnknownDim, in TEST_F()
127 c.Value(InferenceContext::kUnknownDim)); in TEST_F()
297 EXPECT_EQ(InferenceContext::kUnknownDim, c.Value(d)); in TEST_F()
549 {S({2, InferenceContext::kUnknownDim, 2, 1, in TEST_F()
550 InferenceContext::kUnknownDim})}, in TEST_F()
565 EXPECT_EQ(InferenceContext::kUnknownDim, c.Value(out)); in TEST_F()
568 EXPECT_EQ(InferenceContext::kUnknownDim, c.Value(out)); in TEST_F()
572 EXPECT_EQ(InferenceContext::kUnknownDim, c.Value(out)); in TEST_F()
588 EXPECT_EQ(InferenceContext::kUnknownDim, c.Value(out)); in TEST_F()
590 EXPECT_EQ(InferenceContext::kUnknownDim, c.Value(out)); in TEST_F()
[all …]
Dshape_inference_testutil_test.cc89 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, 2)); in TEST()
Dshape_inference_testutil.cc243 dims.push_back(manager->MakeDim(InferenceContext::kUnknownDim)); in MakeShapeFromString()
/external/tensorflow/tensorflow/c/eager/
Dc_api_unified_experimental_graph.cc48 auto& kUnknownDim = shape_inference::InferenceContext::kUnknownDim; variable
73 std::vector<int64> dims(num_dims, kUnknownDim); in Shape()
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/sentencepiece/
Dsentencepiece_tokenizer_op.cc54 tensorflow::shape_inference::InferenceContext::kUnknownDim)); in __anon86abf2d90102()
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner_test.cc873 0, c->MakeShape({1, shape_inference::InferenceContext::kUnknownDim, 3, in __anone6cc7b560c02()
874 shape_inference::InferenceContext::kUnknownDim, 5})); in __anone6cc7b560c02()
884 c->MakeShape({6, shape_inference::InferenceContext::kUnknownDim, 8})); in __anone6cc7b560d02()
1362 shp = ctx->MakeShape({shape_inference::InferenceContext::kUnknownDim, 7}); in TEST_F()

12