Home
last modified time | relevance | path

Searched refs:UnknownDim (Results 1 – 25 of 44) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/boosted_trees/ops/
Dsplit_handler_ops.cc62 c->set_output(0, c->Vector(c->UnknownDim())); in __anon295dc3ce0102()
63 c->set_output(1, c->Vector(c->UnknownDim())); in __anon295dc3ce0102()
64 c->set_output(2, c->Vector(c->UnknownDim())); in __anon295dc3ce0102()
134 c->set_output(0, c->Vector(c->UnknownDim())); in __anon295dc3ce0202()
135 c->set_output(1, c->Vector(c->UnknownDim())); in __anon295dc3ce0202()
136 c->set_output(2, c->Vector(c->UnknownDim())); in __anon295dc3ce0202()
203 c->set_output(0, c->Vector(c->UnknownDim())); in __anon295dc3ce0302()
204 c->set_output(1, c->Vector(c->UnknownDim())); in __anon295dc3ce0302()
205 c->set_output(2, c->Vector(c->UnknownDim())); in __anon295dc3ce0302()
Dstats_accumulator_ops.cc119 c->set_output(1, c->Vector(c->UnknownDim())); in __anone13797730302()
121 c->set_output(3, c->Vector(c->UnknownDim())); in __anone13797730302()
122 c->set_output(4, c->Vector(c->UnknownDim())); in __anone13797730302()
204 c->set_output(2, c->Vector(c->UnknownDim())); in __anone13797730502()
206 c->set_output(4, c->Vector(c->UnknownDim())); in __anone13797730502()
207 c->set_output(5, c->Vector(c->UnknownDim())); in __anone13797730502()
338 c->set_output(1, c->Vector(c->UnknownDim())); in __anone13797730802()
424 c->set_output(2, c->Vector(c->UnknownDim())); in __anone13797730a02()
/external/tensorflow/tensorflow/core/ops/
Dset_ops.cc81 output_rank = c->UnknownDim(); in __anon967df2c80102()
85 c->set_output(0, c->Matrix(c->UnknownDim(), output_rank)); in __anon967df2c80102()
86 c->set_output(1, c->Vector(c->UnknownDim())); in __anon967df2c80102()
126 output_rank_dim = c->UnknownDim(); in __anon967df2c80202()
129 c->set_output(0, c->Matrix(c->UnknownDim(), output_rank_dim)); in __anon967df2c80202()
130 c->set_output(1, c->Vector(c->UnknownDim())); in __anon967df2c80202()
182 output_rank_dim = c->UnknownDim(); in __anon967df2c80302()
185 c->set_output(0, c->Matrix(c->UnknownDim(), output_rank_dim)); in __anon967df2c80302()
186 c->set_output(1, c->Vector(c->UnknownDim())); in __anon967df2c80302()
Dparsing_ops.cc76 c->set_output(output_idx++, c->Matrix(c->UnknownDim(), 2)); in __anon952516500202()
79 c->set_output(output_idx++, c->Vector(c->UnknownDim())); in __anon952516500202()
119 c->set_output(output_idx++, c->Matrix(c->UnknownDim(), 1)); in __anon952516500302()
122 c->set_output(output_idx++, c->Vector(c->UnknownDim())); in __anon952516500302()
183 c->set_output(output_idx++, c->Matrix(c->UnknownDim(), 2)); in __anon952516500402()
186 c->set_output(output_idx++, c->Vector(c->UnknownDim())); in __anon952516500402()
204 c->set_output(output_idx++, c->Matrix(c->UnknownDim(), 3)); in __anon952516500402()
207 c->set_output(output_idx++, c->Vector(c->UnknownDim())); in __anon952516500402()
219 c->Concatenate(c->Matrix(num_examples, c->UnknownDim()), s, &s)); in __anon952516500402()
278 c->set_output(output_idx++, c->Matrix(c->UnknownDim(), 1)); in __anon952516500502()
[all …]
Dimage_ops.cc43 width = c->UnknownDim(); in SetOutputToSizedImage()
44 height = c->UnknownDim(); in SetOutputToSizedImage()
76 channels_dim = c->UnknownDim(); in DecodeImageShapeFn()
130 c->set_output(0, c->Vector(c->UnknownDim())); in NMSShapeFn()
336 TF_RETURN_IF_ERROR(c->ReplaceDim(input, 1, c->UnknownDim(), &input)); in __anon6a71d27f0602()
337 TF_RETURN_IF_ERROR(c->ReplaceDim(input, 2, c->UnknownDim(), &input)); in __anon6a71d27f0602()
371 h = c->UnknownDim(); in __anon6a71d27f0702()
372 w = c->UnknownDim(); in __anon6a71d27f0702()
409 DimensionHandle channels_dim = c->UnknownDim(); in __anon6a71d27f0802()
410 DimensionHandle h = c->UnknownDim(); in __anon6a71d27f0802()
[all …]
Dbatch_ops.cc63 c->ReplaceDim(in_shapes[i], 0, c->UnknownDim(), &out_shapes[i])); in __anon130e45de0102()
69 {c->MakeShape({shape_inference::DimensionOrConstant(c->UnknownDim()), in __anon130e45de0102()
86 c->ReplaceDim(c->input(0), 0, c->UnknownDim(), &out_shape)); in __anon130e45de0202()
Daudio_ops.cc37 channels_dim = c->UnknownDim(); in DecodeWavShapeFn()
49 samples_dim = c->UnknownDim(); in DecodeWavShapeFn()
83 output_length = c->UnknownDim(); in SpectrogramShapeFn()
Dlist_ops.cc230 num_elements = c->MakeShape({c->UnknownDim()}); in __anonebbd57d80602()
267 c->Concatenate(c->MakeShape({c->UnknownDim()}), result, &result)); in TensorListConcatShapeInference()
272 c->set_output(1, c->MakeShape({c->UnknownDim()})); in TensorListConcatShapeInference()
326 TF_RETURN_IF_ERROR(c->Concatenate(c->MakeShape({c->UnknownDim()}), in __anonebbd57d80902()
373 c->set_output(0, c->Vector(c->UnknownDim())); in __anonebbd57d80b02()
Ddecode_proto_ops.cc54 c->Concatenate(input, c->Vector(c->UnknownDim()), &values)); in __anonafeac6530102()
Dragged_conversion_ops.cc64 : c->UnknownDim(); in RaggedTensorToSparseShapeFn()
Dtensor_forest_ops.cc68 shape_inference::DimensionHandle batch_size = c->UnknownDim(); in __anoncf2fc2a90302()
Dragged_math_ops.cc54 DimensionHandle dim = c->UnknownDim(); in RaggedRangeShapeFn()
Dstring_ops.cc315 DimensionHandle num_chars = c->UnknownDim(); in __anonb4ab91930902()
337 DimensionHandle num_chars = c->UnknownDim(); in __anonb4ab91930a02()
Dtpu_configuration_ops.cc167 c->set_output(0, c->Vector(c->UnknownDim())); in __anon5d0267cc0402()
Dsparse_ops.cc222 DimensionHandle output_ind_cols = c->UnknownDim(); in __anona7ffb82a0a02()
269 c->set_output(0, c->Matrix(c->UnknownDim(), 2)); in __anona7ffb82a0b02()
270 c->set_output(1, c->Vector(c->UnknownDim())); in __anona7ffb82a0b02()
/external/tensorflow/tensorflow/contrib/tensor_forest/ops/
Dmodel_ops.cc95 DimensionHandle num_points = c->UnknownDim(); in __anon5b1697710102()
102 c->set_output(0, c->Matrix(num_points, c->UnknownDim())); in __anon5b1697710102()
103 c->set_output(1, c->Vector(c->UnknownDim())); in __anon5b1697710102()
130 DimensionHandle num_points = c->UnknownDim(); in __anon5b1697710202()
177 c->set_output(0, c->Vector(c->UnknownDim())); in __anon5b1697710302()
Dstats_ops.cc104 c->set_output(0, c->Vector(c->UnknownDim())); in __anon58920fcf0102()
/external/tensorflow/tensorflow/contrib/layers/ops/
Dsparse_feature_cross_op.cc36 c->set_output(0, c->Matrix(c->UnknownDim(), 2)); in __anon9c7e7a520102()
37 c->set_output(1, c->Vector(c->UnknownDim())); in __anon9c7e7a520102()
108 c->set_output(0, c->Matrix(c->UnknownDim(), 2)); in __anon9c7e7a520202()
109 c->set_output(1, c->Vector(c->UnknownDim())); in __anon9c7e7a520202()
/external/tensorflow/tensorflow/contrib/ffmpeg/
Ddecode_audio_op.cc182 c->set_output(0, c->Matrix(c->UnknownDim(), c->UnknownDim())); in __anon6f0383490202()
190 c->set_output(0, c->Matrix(c->UnknownDim(), channels)); in __anon6f0383490202()
270 c->set_output(0, c->Matrix(c->UnknownDim(), channels)); in __anon6f0383490302()
272 c->set_output(0, c->Matrix(c->UnknownDim(), c->UnknownDim())); in __anon6f0383490302()
/external/tensorflow/tensorflow/contrib/text/ops/
Dskip_gram_ops.cc46 c->set_output(0, c->Vector(c->UnknownDim())); in __anon1ca0b1550102()
47 c->set_output(1, c->Vector(c->UnknownDim())); in __anon1ca0b1550102()
/external/tensorflow/tensorflow/core/framework/
Dshape_inference.cc313 if (rank == kUnknownRank) return UnknownDim(); in NumElements()
327 return UnknownDim(); in NumElements()
380 dims.push_back(UnknownDim()); in WithRank()
459 *out = UnknownDim(); in Relax()
759 dims[i] = UnknownDim(); in UnknownShapeOfRank()
840 for (int i = 0; i < num_dims; i++) dims.push_back(UnknownDim()); in InternalMakeShapeFromTensor()
971 *out = UnknownDim(); in MakeDimForScalarInput()
988 *out = UnknownDim(); in MakeDimForScalarInputWithNegativeIndexing()
994 *out = UnknownDim(); in MakeDimForScalarInputWithNegativeIndexing()
1020 *out = UnknownDim(); in Divide()
[all …]
Dshape_inference_test.cc1026 auto d1 = c.UnknownDim(); in TEST_F()
1042 auto d1 = c.UnknownDim(); in TEST_F()
1043 auto d2 = c.UnknownDim(); in TEST_F()
1209 TEST_F(ShapeInferenceTest, UnknownDim) { in TEST_F() argument
1214 auto d0 = c.UnknownDim(); in TEST_F()
1215 auto d1 = c.UnknownDim(); in TEST_F()
1392 EXPECT_TRUE(c.Add(d_6, c.UnknownDim(), &out).ok()); in TEST_F()
1442 EXPECT_TRUE(c.Subtract(d_6, c.UnknownDim(), &out).ok()); in TEST_F()
1501 EXPECT_TRUE(c.Multiply(d_6, c.UnknownDim(), &out).ok()); in TEST_F()
1512 EXPECT_FALSE(c.FullyDefined(c.Matrix(c.MakeDim(1), c.UnknownDim()))); in TEST_F()
/external/tensorflow/tensorflow/contrib/image/ops/
Dimage_ops.cc43 width = c->UnknownDim(); in SetOutputToSizedImage()
44 height = c->UnknownDim(); in SetOutputToSizedImage()
/external/tensorflow/tensorflow/contrib/mpi_collectives/ops/
Dmpi_ops.cc109 c->ReplaceDim(c->input(0), 0, c->UnknownDim(), &output)); in __anon107d6f1b0502()
/external/tensorflow/tensorflow/contrib/reduce_slice_ops/ops/
Dreduce_slice_ops.cc30 DimensionHandle dim_axis = c->UnknownDim(); in ReduceSliceShapeFn()

12