| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.h | 144 Status GetWindowedOutputSizeFromDims(InferenceContext* c,
152 Status GetWindowedOutputSizeFromDimsV2(InferenceContext* c,
160 Status UnchangedShape(shape_inference::InferenceContext* c);
163 inline Status UnchangedShapeWithRank(shape_inference::InferenceContext* c, in UnchangedShapeWithRank()
173 shape_inference::InferenceContext* c, int32 rank) { in UnchangedShapeWithRankAtLeast()
181 inline Status UnchangedShapeWithRankAtMost(shape_inference::InferenceContext* c, in UnchangedShapeWithRankAtMost()
190 inline Status NoOutputs(shape_inference::InferenceContext* c) { in NoOutputs()
195 inline Status ScalarShape(shape_inference::InferenceContext* c) { in ScalarShape()
201 inline Status MergeBothInputsShapeFn(InferenceContext* c) { in MergeBothInputsShapeFn()
214 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
85 InferenceContext::InferenceContext( in InferenceContext() function in tensorflow::shape_inference::InferenceContext
136 InferenceContext::InferenceContext( in InferenceContext() function in tensorflow::shape_inference::InferenceContext
152 InferenceContext::~InferenceContext() {} in ~InferenceContext()
154 Status InferenceContext::Run( in Run()
155 const std::function<Status(shape_inference::InferenceContext* c)>& fn) { in Run()
169 Status InferenceContext::set_output(StringPiece output_name, in set_output()
188 Status InferenceContext::input(StringPiece input_name, in input()
[all …]
|
| D | common_shape_fns_test.cc | 65 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, op_def, {S({}), S({10})}, {}, in TEST()
84 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, op_def, {S({})}, {}, {}, {}); in TEST()
91 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, op_def, in TEST()
119 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, op_def, in TEST()
129 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, op_def, in TEST()
139 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()
172 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, op_def, in TEST()
191 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, op_def, in TEST()
[all …]
|
| D | shape_inference_test.cc | 57 void Relax(InferenceContext* c, DimensionHandle d0, DimensionHandle d1, in Relax()
61 void Relax(InferenceContext* c, ShapeHandle s0, ShapeHandle s1, in Relax()
79 InferenceContext c(kVersion, &def, op_def, {S({1, 5}), S({2, 5}), S({1, 3})}, in TEST_F()
116 InferenceContext c(kVersion, &def, MakeOpDef(1, 1), {Unknown()}, {}, {}, {}); in TEST_F()
117 EXPECT_EQ(InferenceContext::kUnknownDim, in TEST_F()
118 c.Value(InferenceContext::kUnknownDim)); in TEST_F()
131 InferenceContext c(kVersion, &def, MakeOpDef(1, 2), {S({1})}, {}, {}, {}); in TEST_F()
135 auto fn = [](InferenceContext* c) { in TEST_F()
146 auto fn = [](InferenceContext* c) { in TEST_F()
170 InferenceContext c(kVersion, &def, MakeOpDef(1, 2), {S({1, 2, 3})}, {}, {}, in TEST_F()
[all …]
|
| D | shape_inference_testutil_test.cc | 40 .SetShapeFn([](InferenceContext* c) { return (*global_fn_ptr)(c); }); in __anon3551c0b90202()
47 .SetShapeFn([](InferenceContext* c) { return (*global_fn_ptr)(c); }); in __anon3551c0b90302()
69 auto fn_copy_input_0 = [](InferenceContext* c) { in TEST()
73 auto fn_copy_input_2 = [](InferenceContext* c) { in TEST()
77 auto fn_output_unknown_shapes = [](InferenceContext* c) { in TEST()
83 auto fn_output_1_2 = [](InferenceContext* c) { in TEST()
87 auto fn_output_u_2 = [](InferenceContext* c) { in TEST()
88 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, 2)); in TEST()
139 auto fn = [](InferenceContext* c) { in TEST()
|
| D | shape_inference.h | 40 class InferenceContext; variable
51 friend class InferenceContext; variable
71 friend class InferenceContext; variable
92 friend class InferenceContext; variable
112 friend class InferenceContext; variable
158 class InferenceContext {
173 InferenceContext(int graph_def_version, const NodeDef* node_def,
192 InferenceContext(
212 InferenceContext(
221 ~InferenceContext();
[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 __anon4c2bcc0f0102()
126 .SetShapeFn([](InferenceContext* c) { in __anon4c2bcc0f0202()
138 .SetShapeFn([](InferenceContext* c) { in __anon4c2bcc0f0302()
152 .SetShapeFn([](InferenceContext* c) { in __anon4c2bcc0f0402()
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 __anone6e195410202()
75 .SetShapeFn([](InferenceContext* c) { in __anone6e195410302()
79 0, c->Matrix(InferenceContext::kUnknownDim, c->Dim(a_shape, 0))); in __anone6e195410302()
80 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in __anone6e195410302()
95 .SetShapeFn([](InferenceContext* c) { in __anone6e195410402()
[all …]
|
| D | spectral_ops.cc | 24 using shape_inference::InferenceContext;
30 .SetShapeFn([](InferenceContext* c) { in __anon41aea5d10102()
37 .SetShapeFn([](InferenceContext* c) { in __anon41aea5d10202()
44 .SetShapeFn([](InferenceContext* c) { in __anon41aea5d10302()
51 .SetShapeFn([](InferenceContext* c) { in __anon41aea5d10402()
58 .SetShapeFn([](InferenceContext* c) { in __anon41aea5d10502()
65 .SetShapeFn([](InferenceContext* c) { in __anon41aea5d10602()
69 Status RFFTShape(InferenceContext* c, const bool forward, const int rank) { in RFFTShape()
109 .SetShapeFn([](InferenceContext* c) { return RFFTShape(c, true, 1); }); in __anon41aea5d10702()
115 .SetShapeFn([](InferenceContext* c) { return RFFTShape(c, false, 1); }); in __anon41aea5d10802()
[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()
138 .SetShapeFn([](InferenceContext* c) { in __anonc970e8b60202()
165 .SetShapeFn([](InferenceContext* c) { in __anonc970e8b60302()
[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 __anonec23b74b0202()
93 .SetShapeFn([](InferenceContext* c) { in __anonec23b74b0302()
137 .SetShapeFn([](InferenceContext* c) { in __anonec23b74b0402()
150 .SetShapeFn([](InferenceContext* c) { in __anonec23b74b0502()
173 .SetShapeFn([](InferenceContext* c) { in __anonec23b74b0602()
199 .SetShapeFn([](InferenceContext* c) { in __anonec23b74b0702()
[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 __anona2f81a440202()
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 __anona2f81a440502()
338 .SetShapeFn([](InferenceContext* c) { in __anona2f81a440602()
341 n_shape = c->Vector(InferenceContext::kUnknownDim); in __anona2f81a440602()
366 .SetShapeFn([](InferenceContext* c) { in __anona2f81a440702()
[all …]
|
| D | lookup_ops.cc | 25 using shape_inference::InferenceContext;
31 Status TwoElementVectorInputsAndScalarOutputs(InferenceContext* c) { in TwoElementVectorInputsAndScalarOutputs()
44 Status ScalarAndTwoElementVectorInputsAndScalarOutputs(InferenceContext* c) { in ScalarAndTwoElementVectorInputsAndScalarOutputs()
58 Status TwoElementOutput(InferenceContext* c) { in TwoElementOutput()
63 Status ScalarOutput(InferenceContext* c) { in ScalarOutput()
76 .SetShapeFn([](InferenceContext* c) { in __anonc8e9eead0202()
96 .SetShapeFn([](InferenceContext* c) { in __anonc8e9eead0302()
115 .SetShapeFn([](InferenceContext* c) { in __anonc8e9eead0402()
131 .SetShapeFn([](InferenceContext* c) { in __anonc8e9eead0502()
155 .SetShapeFn([](InferenceContext* c) { in __anonc8e9eead0602()
[all …]
|
| D | nn_ops.cc | 27 using shape_inference::InferenceContext;
32 Status FractionalPoolShapeFn(InferenceContext* c) { in FractionalPoolShapeFn()
88 .SetShapeFn([](InferenceContext* c) { in __anon4650176f0202()
109 .SetShapeFn([](InferenceContext* c) { in __anon4650176f0302()
141 .SetShapeFn([](InferenceContext* c) { in __anon4650176f0402()
288 .SetShapeFn([](InferenceContext* c) { in __anon4650176f0502()
310 .SetShapeFn([](InferenceContext* c) { in __anon4650176f0602()
320 Status CommonFusedConvCalculations(InferenceContext* c, bool has_resize) { in CommonFusedConvCalculations()
440 .SetShapeFn([](InferenceContext* c) { in __anon4650176f0802()
453 .SetShapeFn([](InferenceContext* c) { in __anon4650176f0902()
[all …]
|
| D | array_ops.cc | 28 using shape_inference::InferenceContext;
33 Status GetAxisForPackAndUnpack(InferenceContext* c, int32 rank_after_pack, in GetAxisForPackAndUnpack()
56 Status PadKnown(InferenceContext* c, ShapeHandle input, in PadKnown()
73 Status PadShapeFn(InferenceContext* c) { in PadShapeFn()
113 Status TransposeShapeFn(InferenceContext* c) { in TransposeShapeFn()
171 Status SetOutputShapeForReshape(InferenceContext* c) { in SetOutputShapeForReshape()
233 .SetShapeFn([](InferenceContext* c) { in __anondb9326b20202()
272 .SetShapeFn([](InferenceContext* c) { in __anondb9326b20302()
309 .SetShapeFn([](InferenceContext* c) { in __anondb9326b20402()
343 .SetShapeFn([](InferenceContext* c) { return Status::OK(); }); in __anondb9326b20502()
[all …]
|
| D | resource_variable_ops.cc | 22 using ::tensorflow::shape_inference::InferenceContext;
30 Status ValidateVariableResourceHandle(InferenceContext* c, in ValidateVariableResourceHandle()
51 Status ReadVariableShapeFn(InferenceContext* c) { in ReadVariableShapeFn()
67 .SetShapeFn([](InferenceContext* c) { in __anon4d581ac00202()
93 Status CreateAssignShapeFn(InferenceContext* c) { in CreateAssignShapeFn()
127 Status VariableShapeShapeFn(InferenceContext* c) { in VariableShapeShapeFn()
134 : InferenceContext::kUnknownDim; in VariableShapeShapeFn()
152 .SetShapeFn([](InferenceContext* c) { in __anon4d581ac00302()
176 .SetShapeFn([](InferenceContext* c) { in __anon4d581ac00402()
198 .SetShapeFn([](InferenceContext* c) { in __anon4d581ac00502()
[all …]
|
| D | math_ops.cc | 24 using shape_inference::InferenceContext;
34 .SetShapeFn([](InferenceContext* c) { in __anon57f5a5dd0102()
71 .SetShapeFn([](InferenceContext* c) { in __anon57f5a5dd0202()
515 .SetShapeFn([](InferenceContext* c) { in __anon57f5a5dd0302()
625 .SetShapeFn([](InferenceContext* c) { in __anon57f5a5dd0402()
782 Status ArgOpShape(shape_inference::InferenceContext* c) { in ArgOpShape()
859 Status SegmentReductionShapeFn(InferenceContext* c) { in SegmentReductionShapeFn()
870 c->Concatenate(c->Vector(InferenceContext::kUnknownDim), subshape, &out)); in SegmentReductionShapeFn()
875 Status SparseSegmentReductionShapeFn(InferenceContext* c) { in SparseSegmentReductionShapeFn()
894 c->Concatenate(c->Vector(InferenceContext::kUnknownDim), subshape, &out)); in SparseSegmentReductionShapeFn()
[all …]
|
| D | linalg_ops.cc | 23 using shape_inference::InferenceContext;
29 Status MakeBatchSquareMatrix(InferenceContext* c, ShapeHandle input, in MakeBatchSquareMatrix()
43 Status BatchUnchangedSquareShapeFn(InferenceContext* c) { in BatchUnchangedSquareShapeFn()
52 Status MatrixSolveShapeFn(InferenceContext* c, bool square) { in MatrixSolveShapeFn()
90 Status SelfAdjointEigV2ShapeFn(InferenceContext* c) { in SelfAdjointEigV2ShapeFn()
117 Status QrShapeFn(InferenceContext* c) { in QrShapeFn()
148 Status SvdShapeFn(InferenceContext* c) { in SvdShapeFn()
193 .SetShapeFn([](InferenceContext* c) { in __anon8ebee8ea0202()
212 .SetShapeFn([](InferenceContext* c) { in __anon8ebee8ea0302()
267 .SetShapeFn([](InferenceContext* c) { in __anon8ebee8ea0402()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 32 using shape_inference::InferenceContext;
72 if (InferenceContext::RankKnown(*result)) { in Merge()
77 if (InferenceContext::RankKnown(h1)) { in Merge()
91 if (!InferenceContext::ValueKnown(d)) { in ExtractValue()
95 int64 val = InferenceContext::Value(d); in ExtractValue()
111 const int64 dim1 = InferenceContext::Value(d1); in Merge()
112 const int64 dim2 = InferenceContext::Value(d2); in Merge()
391 InferenceContext* GetContext(const Node* node) { in GetContext()
400 InferenceContext* ctx = GetContext(node); in SetUnknownShape()
444 InferenceContext* ctx = shape_refiner_->GetContext(node); in OutputAsIntersection()
[all …]
|
| /external/tensorflow/tensorflow/contrib/libsvm/ops/ |
| D | libsvm_ops.cc | 22 using shape_inference::InferenceContext;
33 .SetShapeFn([](InferenceContext* c) { in __anonccef2bc40102()
36 c->set_output(1, c->Matrix(InferenceContext::kUnknownDim, in __anonccef2bc40102()
37 InferenceContext::kUnknownDim)); in __anonccef2bc40102()
38 c->set_output(2, c->Vector(InferenceContext::kUnknownDim)); in __anonccef2bc40102()
39 c->set_output(3, c->Vector(InferenceContext::kUnknownDim)); in __anonccef2bc40102()
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/ops/ |
| D | cudnn_rnn_ops.cc | 71 using shape_inference::InferenceContext;
87 .SetShapeFn([](InferenceContext* c) { in __anone2c774540202()
141 .SetShapeFn([](InferenceContext* c) { in __anone2c774540302()
200 .SetShapeFn([](InferenceContext* c) { in __anone2c774540402()
247 .SetShapeFn([](InferenceContext* c) { in __anone2c774540502()
254 c->set_output(i, c->Matrix(InferenceContext::kUnknownDim, in __anone2c774540502()
255 InferenceContext::kUnknownDim)); in __anone2c774540502()
259 c->set_output(num_params + i, c->Vector(InferenceContext::kUnknownDim)); in __anone2c774540502()
289 .SetShapeFn([](InferenceContext* c) { in __anone2c774540602()
290 c->set_output(0, c->Vector(InferenceContext::kUnknownDim)); in __anone2c774540602()
|
| /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()
107 .SetShapeFn([](InferenceContext* c) { in __anon5dcf83ad0102()
109 {c->Vector(InferenceContext::kUnknownDim)}); in __anon5dcf83ad0102()
|
| D | stats_accumulator_ops.cc | 23 using shape_inference::InferenceContext;
39 .SetShapeFn([](InferenceContext* c) { in __anon72691d0a0102()
61 .SetShapeFn([](InferenceContext* c) { in __anon72691d0a0202()
113 .SetShapeFn([](InferenceContext* c) { in __anon72691d0a0302()
150 .SetShapeFn([](InferenceContext* c) { in __anon72691d0a0402()
197 .SetShapeFn([](InferenceContext* c) { in __anon72691d0a0502()
253 .SetShapeFn([](InferenceContext* c) { in __anon72691d0a0602()
279 .SetShapeFn([](InferenceContext* c) { in __anon72691d0a0702()
331 .SetShapeFn([](InferenceContext* c) { in __anon72691d0a0802()
369 .SetShapeFn([](InferenceContext* c) { in __anon72691d0a0902()
[all …]
|
| /external/tensorflow/tensorflow/contrib/mpi_collectives/ops/ |
| D | mpi_ops.cc | 35 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon3b0c9a320102()
50 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon3b0c9a320202()
65 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon3b0c9a320302()
82 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon3b0c9a320402()
106 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon3b0c9a320502()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner_test.cc | 48 bool SameDefinedShape(shape_inference::InferenceContext* c, in SameDefinedShape()
55 shape_inference::InferenceContext* c, in IsUpdatedShapesOrTypes()
68 shape_inference::InferenceContext* ctx = M.GetContext(OP.node()); \
74 shape_inference::InferenceContext* ctx = M.GetContext(OP.node()); \
83 shape_inference::InferenceContext* ctx = M.GetContext(OP.node()); \
294 shape_inference::InferenceContext* ctx = m.GetContext(shape_v2); in TEST_F()
308 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anone6cc7b560402()
342 shape_inference::InferenceContext* ctx = m.GetContext(node); in TEST_F()
353 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anone6cc7b560602()
372 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anone6cc7b560702()
[all …]
|