| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.h | 153 Status GetWindowedOutputSizeFromDims(InferenceContext* c,
164 InferenceContext* c, DimensionHandle input_size,
170 Status UnchangedShape(shape_inference::InferenceContext* c);
173 inline Status UnchangedShapeWithRank(shape_inference::InferenceContext* c, in UnchangedShapeWithRank()
183 shape_inference::InferenceContext* c, int32 rank) { in UnchangedShapeWithRankAtLeast()
191 inline Status UnchangedShapeWithRankAtMost(shape_inference::InferenceContext* c, in UnchangedShapeWithRankAtMost()
200 inline Status NoOutputs(shape_inference::InferenceContext* c) { in NoOutputs()
205 inline Status ScalarShape(shape_inference::InferenceContext* c) { in ScalarShape()
211 inline Status MergeBothInputsShapeFn(InferenceContext* c) { in MergeBothInputsShapeFn()
224 shape_inference::InferenceContext* context);
[all …]
|
| D | shape_inference.cc | 29 constexpr int32 InferenceContext::kUnknownRank;
30 constexpr int64 InferenceContext::kUnknownDim;
32 InferenceContext::InferenceContext( in InferenceContext() function in tensorflow::shape_inference::InferenceContext
87 InferenceContext::InferenceContext( in InferenceContext() function in tensorflow::shape_inference::InferenceContext
140 InferenceContext::InferenceContext( in InferenceContext() function in tensorflow::shape_inference::InferenceContext
156 InferenceContext::~InferenceContext() {} in ~InferenceContext()
158 Status InferenceContext::Run( in Run()
159 const std::function<Status(shape_inference::InferenceContext* c)>& fn) { in Run()
175 Status InferenceContext::set_output(StringPiece output_name, in set_output()
194 Status InferenceContext::input(StringPiece input_name, in input()
[all …]
|
| D | common_shape_fns_test.cc | 66 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, op_def, {S({}), S({10})}, {}, in TEST()
85 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, op_def, {S({})}, {}, {}, {}); in TEST()
92 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, op_def, in TEST()
120 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, op_def, in TEST()
130 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, op_def, in TEST()
140 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, op_def, {S({2}), S({3, 4})}, in TEST()
150 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, op_def, in TEST()
160 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, op_def, in TEST()
171 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, op_def, in TEST()
189 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, op_def, in TEST()
[all …]
|
| D | shape_inference_testutil_test.cc | 42 .SetShapeFn([](InferenceContext* c) { return (*global_fn_ptr)(c); }); in __anon968f2bc20202()
49 .SetShapeFn([](InferenceContext* c) { return (*global_fn_ptr)(c); }); in __anon968f2bc20302()
71 auto fn_copy_input_0 = [](InferenceContext* c) { in TEST()
75 auto fn_copy_input_2 = [](InferenceContext* c) { in TEST()
79 auto fn_output_unknown_shapes = [](InferenceContext* c) { in TEST()
85 auto fn_output_1_2 = [](InferenceContext* c) { in TEST()
89 auto fn_output_u_2 = [](InferenceContext* c) { in TEST()
90 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, 2)); in TEST()
141 auto fn = [](InferenceContext* c) { in TEST()
|
| D | shape_inference_test.cc | 58 void Relax(InferenceContext* c, DimensionHandle d0, DimensionHandle d1, in Relax()
62 void Relax(InferenceContext* c, ShapeHandle s0, ShapeHandle s1, in Relax()
80 InferenceContext c(kVersion, &def, op_def, {S({1, 5}), S({2, 5}), S({1, 3})}, in TEST_F()
117 InferenceContext c(kVersion, &def, MakeOpDef(1, 1), {Unknown()}, {}, {}, {}); in TEST_F()
118 EXPECT_EQ(InferenceContext::kUnknownDim, in TEST_F()
119 c.Value(InferenceContext::kUnknownDim)); in TEST_F()
132 InferenceContext c(kVersion, &def, MakeOpDef(1, 2), {S({1})}, {}, {}, {}); in TEST_F()
136 auto fn = [](InferenceContext* c) { in TEST_F()
147 auto fn = [](InferenceContext* c) { in TEST_F()
170 InferenceContext c(kVersion, &def, MakeOpDef(1, 2), {S({1, 2, 3})}, {}, {}, in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | training_ops.cc | 22 using shape_inference::InferenceContext;
25 static ShapeHandle ShapeOrHandleShape(InferenceContext* c, int input) { in ShapeOrHandleShape()
37 static Status HandleGradAndIndicesInputs(InferenceContext* c, bool sparse, in HandleGradAndIndicesInputs()
59 static Status ApplyGradientDescentShapeFn(InferenceContext* c) { in ApplyGradientDescentShapeFn()
87 static Status ApplyProximalGradientDescentShapeFn(InferenceContext* c, in ApplyProximalGradientDescentShapeFn()
111 .SetShapeFn([](InferenceContext* c) { in __anoncd5e4d380102()
126 .SetShapeFn([](InferenceContext* c) { in __anoncd5e4d380202()
138 .SetShapeFn([](InferenceContext* c) { in __anoncd5e4d380302()
152 .SetShapeFn([](InferenceContext* c) { in __anoncd5e4d380402()
156 static Status ApplyAdadeltaShapeFn(InferenceContext* c, bool sparse) { in ApplyAdadeltaShapeFn()
[all …]
|
| D | sparse_ops.cc | 23 using shape_inference::InferenceContext;
28 Status SparseSparseMinOrMaxShapeFn(InferenceContext* c) { in SparseSparseMinOrMaxShapeFn()
36 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()
52 .SetShapeFn([](InferenceContext* c) { in __anona7ffb82a0202()
75 .SetShapeFn([](InferenceContext* c) { in __anona7ffb82a0302()
79 0, c->Matrix(InferenceContext::kUnknownDim, c->Dim(a_shape, 0))); in __anona7ffb82a0302()
80 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in __anona7ffb82a0302()
95 .SetShapeFn([](InferenceContext* c) { in __anona7ffb82a0402()
[all …]
|
| D | spectral_ops.cc | 24 using shape_inference::InferenceContext;
31 .SetShapeFn([](InferenceContext* c) { in __anonc2e126fa0102()
39 .SetShapeFn([](InferenceContext* c) { in __anonc2e126fa0202()
47 .SetShapeFn([](InferenceContext* c) { in __anonc2e126fa0302()
55 .SetShapeFn([](InferenceContext* c) { in __anonc2e126fa0402()
63 .SetShapeFn([](InferenceContext* c) { in __anonc2e126fa0502()
71 .SetShapeFn([](InferenceContext* c) { in __anonc2e126fa0602()
75 Status RFFTShape(InferenceContext* c, const bool forward, const int rank) { in RFFTShape()
115 .SetShapeFn([](InferenceContext* c) { return RFFTShape(c, true, 1); }); in __anonc2e126fa0702()
121 .SetShapeFn([](InferenceContext* c) { return RFFTShape(c, false, 1); }); in __anonc2e126fa0802()
[all …]
|
| D | lookup_ops.cc | 25 using shape_inference::InferenceContext;
32 Status TwoElementVectorInputsAndScalarOutputs(InferenceContext* c) { in TwoElementVectorInputsAndScalarOutputs()
45 Status ScalarAndTwoElementVectorInputsAndScalarOutputs(InferenceContext* c) { in ScalarAndTwoElementVectorInputsAndScalarOutputs()
59 Status TwoElementOutput(InferenceContext* c) { in TwoElementOutput()
64 Status ScalarOutput(InferenceContext* c) { in ScalarOutput()
77 .SetShapeFn([](InferenceContext* c) { in __anon8a0811960202()
90 Status ValidateTableResourceHandle(InferenceContext* c, ShapeHandle keys, in ValidateTableResourceHandle()
165 .SetShapeFn([](InferenceContext* c) { in __anon8a0811960302()
193 .SetShapeFn([](InferenceContext* c) { in __anon8a0811960402()
209 .SetShapeFn([](InferenceContext* c) { in __anon8a0811960502()
[all …]
|
| D | image_ops.cc | 23 using shape_inference::InferenceContext;
30 Status SetOutputToSizedImage(InferenceContext* c, DimensionHandle batch_dim, in SetOutputToSizedImage()
62 Status ResizeShapeFn(InferenceContext* c) { in ResizeShapeFn()
69 Status DecodeImageShapeFn(InferenceContext* c) { in DecodeImageShapeFn()
85 c->set_output(0, c->MakeShape({InferenceContext::kUnknownDim, in DecodeImageShapeFn()
86 InferenceContext::kUnknownDim, channels_dim})); in DecodeImageShapeFn()
90 Status EncodeImageShapeFn(InferenceContext* c) { in EncodeImageShapeFn()
97 Status ColorspaceShapeFn(InferenceContext* c) { in ColorspaceShapeFn()
111 Status NMSShapeFn(InferenceContext* c) { in NMSShapeFn()
134 Status CombinedNMSShapeFn(InferenceContext* c) { in CombinedNMSShapeFn()
[all …]
|
| D | io_ops.cc | 24 using shape_inference::InferenceContext;
29 Status ScalarInputsAndOutputs(InferenceContext* c) { in ScalarInputsAndOutputs()
40 Status TwoElementVectorAndScalarOutputs(InferenceContext* c) { in TwoElementVectorAndScalarOutputs()
53 Status TwoElementOutput(InferenceContext* c) { in TwoElementOutput()
67 .SetShapeFn([](InferenceContext* c) { in __anone43acdb40202()
93 .SetShapeFn([](InferenceContext* c) { in __anone43acdb40302()
137 .SetShapeFn([](InferenceContext* c) { in __anone43acdb40402()
150 .SetShapeFn([](InferenceContext* c) { in __anone43acdb40502()
173 .SetShapeFn([](InferenceContext* c) { in __anone43acdb40602()
199 .SetShapeFn([](InferenceContext* c) { in __anone43acdb40702()
[all …]
|
| D | string_ops.cc | 31 class InferenceContext;
35 using shape_inference::InferenceContext;
44 .SetShapeFn([](InferenceContext* c) { in __anonb4ab91930102()
64 .SetShapeFn([](InferenceContext* c) { in __anonb4ab91930202()
124 .SetShapeFn([](InferenceContext* c) { in __anonb4ab91930302()
148 .SetShapeFn([](InferenceContext* c) { in __anonb4ab91930402()
180 .SetShapeFn([](InferenceContext* c) { in __anonb4ab91930502()
185 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, 2)); in __anonb4ab91930502()
186 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in __anonb4ab91930502()
198 .SetShapeFn([](InferenceContext* c) { in __anonb4ab91930602()
[all …]
|
| D | data_flow_ops.cc | 24 using shape_inference::InferenceContext;
29 Status DequeueManyV2Shape(InferenceContext* c, ShapeHandle n_shape) { in DequeueManyV2Shape()
54 .SetShapeFn([](InferenceContext* c) { in __anon4a7ac08d0202()
91 Status DynamicStitchShapeFunction(InferenceContext* c) { in DynamicStitchShapeFunction()
161 Status TwoElementVectorInputsAndScalarOutputs(InferenceContext* c) { in TwoElementVectorInputsAndScalarOutputs()
174 Status TwoElementOutput(InferenceContext* c) { in TwoElementOutput()
312 .SetShapeFn([](InferenceContext* c) { in __anon4a7ac08d0502()
338 .SetShapeFn([](InferenceContext* c) { in __anon4a7ac08d0602()
341 n_shape = c->Vector(InferenceContext::kUnknownDim); in __anon4a7ac08d0602()
366 .SetShapeFn([](InferenceContext* c) { in __anon4a7ac08d0702()
[all …]
|
| D | nn_ops.cc | 27 using shape_inference::InferenceContext;
32 Status FractionalPoolShapeFn(InferenceContext* c) { in FractionalPoolShapeFn()
88 .SetShapeFn([](InferenceContext* c) { in __anon3e672dd80202()
109 .SetShapeFn([](InferenceContext* c) { in __anon3e672dd80302()
141 .SetShapeFn([](InferenceContext* c) { in __anon3e672dd80402()
290 .SetShapeFn([](InferenceContext* c) { in __anon3e672dd80502()
313 .SetShapeFn([](InferenceContext* c) { in __anon3e672dd80602()
345 Status CommonFusedConvCalculations(InferenceContext* c, bool has_resize) { in CommonFusedConvCalculations()
465 .SetShapeFn([](InferenceContext* c) { in __anon3e672dd80802()
478 .SetShapeFn([](InferenceContext* c) { in __anon3e672dd80902()
[all …]
|
| D | array_ops.cc | 28 using shape_inference::InferenceContext;
34 Status GetAxisForPackAndUnpack(InferenceContext* c, int32 rank_after_pack, in GetAxisForPackAndUnpack()
57 Status PadKnown(InferenceContext* c, ShapeHandle input, in PadKnown()
74 Status PadShapeFn(InferenceContext* c) { in PadShapeFn()
114 Status TransposeShapeFn(InferenceContext* c) { in TransposeShapeFn()
180 Status SetOutputShapeForReshape(InferenceContext* c) { in SetOutputShapeForReshape()
284 .SetShapeFn([](InferenceContext* c) { in __anon7c94107b0202()
323 .SetShapeFn([](InferenceContext* c) { in __anon7c94107b0302()
400 .SetShapeFn([](InferenceContext* c) { in __anon7c94107b0402()
414 .SetShapeFn([](InferenceContext* c) { in __anon7c94107b0502()
[all …]
|
| D | cudnn_rnn_ops.cc | 37 using shape_inference::InferenceContext;
53 .SetShapeFn([](InferenceContext* c) { in __anone1d60d010202()
82 .SetShapeFn([](InferenceContext* c) { in __anone1d60d010302()
125 .SetShapeFn([](InferenceContext* c) { in __anone1d60d010402()
171 .SetShapeFn([](InferenceContext* c) { in __anone1d60d010502()
221 .SetShapeFn([](InferenceContext* c) { in __anone1d60d010602()
258 .SetShapeFn([](InferenceContext* c) { in __anone1d60d010702()
297 .SetShapeFn([](InferenceContext* c) { in __anone1d60d010802()
324 .SetShapeFn([](InferenceContext* c) { in __anone1d60d010902()
331 c->set_output(i, c->Matrix(InferenceContext::kUnknownDim, in __anone1d60d010902()
[all …]
|
| D | resource_variable_ops.cc | 24 using ::tensorflow::shape_inference::InferenceContext;
33 InferenceContext* c, std::vector<ShapeAndType>* shape_and_type) { in ValidateVariableResourceHandle()
52 Status ReadVariableShapeFn(InferenceContext* c) { in ReadVariableShapeFn()
65 Status ReadVariablesShapeFn(InferenceContext* c) { in ReadVariablesShapeFn()
104 .SetShapeFn([](InferenceContext* c) { in __anon3f9cfa090202()
126 .SetShapeFn([](InferenceContext* c) { in __anon3f9cfa090302()
187 Status CreateAssignShapeFn(InferenceContext* c) { in CreateAssignShapeFn()
235 Status VariableShapeShapeFn(InferenceContext* c) { in VariableShapeShapeFn()
243 : InferenceContext::kUnknownDim; in VariableShapeShapeFn()
262 .SetShapeFn([](InferenceContext* c) { in __anon3f9cfa090402()
[all …]
|
| D | boosted_trees_ops.cc | 27 using shape_inference::InferenceContext;
35 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anone3ce78ae0102()
56 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anone3ce78ae0202()
104 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anone3ce78ae0302()
116 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anone3ce78ae0402()
131 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anone3ce78ae0502()
151 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anone3ce78ae0602()
192 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anone3ce78ae0702()
219 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anone3ce78ae0802()
241 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anone3ce78ae0902()
[all …]
|
| D | dataset_ops.cc | 108 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anondee5e0a20102()
120 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anondee5e0a20202()
132 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anondee5e0a20302()
170 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anondee5e0a20402()
239 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anondee5e0a20502()
255 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anondee5e0a20602()
269 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anondee5e0a20702()
285 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anondee5e0a20802()
307 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anondee5e0a20902()
324 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anondee5e0a20a02()
[all …]
|
| D | list_ops.cc | 26 shape_inference::InferenceContext* c, in VerifyHandleData()
57 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonebbd57d80202()
75 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonebbd57d80302()
112 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonebbd57d80402()
165 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonebbd57d80502()
202 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonebbd57d80602()
241 shape_inference::InferenceContext* c, in TensorListConcatShapeInference()
282 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonebbd57d80702()
299 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonebbd57d80802()
313 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonebbd57d80902()
[all …]
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/ops/ |
| D | prediction_ops.cc | 24 using shape_inference::InferenceContext;
26 static Status ApplyGradientTreesPredictionShapeFn(InferenceContext* c) { in ApplyGradientTreesPredictionShapeFn()
36 c->set_output(0, {c->Matrix(InferenceContext::kUnknownDim, in ApplyGradientTreesPredictionShapeFn()
43 static Status ApplyGradientTreesPredictionVerboseShapeFn(InferenceContext* c) { in ApplyGradientTreesPredictionVerboseShapeFn()
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()
177 .SetShapeFn([](InferenceContext* c) { in __anonbe66d2360102()
179 {c->Vector(InferenceContext::kUnknownDim)}); in __anonbe66d2360102()
|
| /external/tensorflow/tensorflow/contrib/libsvm/ops/ |
| D | libsvm_ops.cc | 22 using shape_inference::InferenceContext;
33 .SetShapeFn([](InferenceContext* c) { in __anonbd00c5ed0102()
36 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/core/grappler/costs/ |
| D | graph_properties.cc | 45 using shape_inference::InferenceContext;
86 if (InferenceContext::RankKnown(*result)) { in Merge()
91 if (InferenceContext::RankKnown(h1)) { in Merge()
105 if (!InferenceContext::ValueKnown(d)) { in ExtractValue()
109 int64 val = InferenceContext::Value(d); in ExtractValue()
125 const int64 dim1 = InferenceContext::Value(d1); in Merge()
126 const int64 dim2 = InferenceContext::Value(d2); in Merge()
356 InferenceContext* ic, const ShapeHandle& shape, in IsShapeFullyDefinedIntegerVectorOrScalar()
368 TensorProto MakeTensorProtoFromShape(InferenceContext* ic, in MakeTensorProtoFromShape()
391 NodeDef MakeConstNodeDefFromTensorProto(InferenceContext* ic, in MakeConstNodeDefFromTensorProto()
[all …]
|
| /external/tensorflow/tensorflow/contrib/mpi_collectives/ops/ |
| D | mpi_ops.cc | 35 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon107d6f1b0102()
50 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon107d6f1b0202()
65 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon107d6f1b0302()
82 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon107d6f1b0402()
106 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon107d6f1b0502()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.h | 38 std::unique_ptr<shape_inference::InferenceContext> ic, const Node* node) in ExtendedInferenceContext()
57 shape_inference::InferenceContext* get_context() { in get_context()
71 std::unique_ptr<shape_inference::InferenceContext> inference_context_;
129 shape_inference::InferenceContext* GetContext(const Node* node) const { in GetContext()
191 static bool SameDefinedShape(shape_inference::InferenceContext* c,
198 shape_inference::InferenceContext* c,
251 Status ConstantPartialShape(shape_inference::InferenceContext* target_context,
257 shape_inference::InferenceContext* ctx,
|