| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.h | 30 Status GetWindowedOutputSizeFromDims(InferenceContext* c,
41 InferenceContext* c, DimensionHandle input_size,
47 Status UnchangedShape(shape_inference::InferenceContext* c);
50 inline Status UnchangedShapeWithRank(shape_inference::InferenceContext* c, in UnchangedShapeWithRank()
60 shape_inference::InferenceContext* c, int32 rank) { in UnchangedShapeWithRankAtLeast()
68 inline Status UnchangedShapeWithRankAtMost(shape_inference::InferenceContext* c, in UnchangedShapeWithRankAtMost()
77 inline Status NoOutputs(shape_inference::InferenceContext* c) { in NoOutputs()
82 inline Status ScalarShape(shape_inference::InferenceContext* c) { in ScalarShape()
88 inline Status MergeBothInputsShapeFn(InferenceContext* c) { in MergeBothInputsShapeFn()
96 Status DatasetIteratorShape(shape_inference::InferenceContext* c);
[all …]
|
| D | shape_inference.cc | 29 constexpr int32 InferenceContext::kUnknownRank;
30 constexpr int64 InferenceContext::kUnknownDim;
33 InferenceContext::InferenceContext( in InferenceContext() function in tensorflow::shape_inference::InferenceContext
85 InferenceContext::InferenceContext( in InferenceContext() function in tensorflow::shape_inference::InferenceContext
100 InferenceContext::~InferenceContext() {} in ~InferenceContext()
102 Status InferenceContext::Run( in Run()
103 const std::function<Status(shape_inference::InferenceContext* c)>& fn) { in Run()
118 Status InferenceContext::set_output(StringPiece output_name, in set_output()
138 Status InferenceContext::input(StringPiece input_name, in input()
152 Status InferenceContext::output(StringPiece output_name, in output()
[all …]
|
| D | shape_inference_testutil_test.cc | 41 .SetShapeFn([](InferenceContext* c) { return (*global_fn_ptr)(c); }); in __anonabaab5e00202()
48 .SetShapeFn([](InferenceContext* c) { return (*global_fn_ptr)(c); }); in __anonabaab5e00302()
70 auto fn_copy_input_0 = [](InferenceContext* c) { in TEST()
74 auto fn_copy_input_2 = [](InferenceContext* c) { in TEST()
78 auto fn_output_unknown_shapes = [](InferenceContext* c) { in TEST()
84 auto fn_output_1_2 = [](InferenceContext* c) { in TEST()
88 auto fn_output_u_2 = [](InferenceContext* c) { in TEST()
89 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, 2)); in TEST()
140 auto fn = [](InferenceContext* c) { in TEST()
|
| 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, {S({1, 23, 4, 4, 2})}, 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 …]
|
| /external/tensorflow/tensorflow/core/ops/ |
| 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 __anon36a8fc880202()
75 .SetShapeFn([](InferenceContext* c) { in __anon36a8fc880302()
79 0, c->Matrix(InferenceContext::kUnknownDim, c->Dim(a_shape, 0))); in __anon36a8fc880302()
80 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in __anon36a8fc880302()
95 .SetShapeFn([](InferenceContext* c) { in __anon36a8fc880402()
[all …]
|
| D | spectral_ops.cc | 24 using shape_inference::InferenceContext;
31 .SetShapeFn([](InferenceContext* c) { in __anona0ecfad80102()
39 .SetShapeFn([](InferenceContext* c) { in __anona0ecfad80202()
47 .SetShapeFn([](InferenceContext* c) { in __anona0ecfad80302()
55 .SetShapeFn([](InferenceContext* c) { in __anona0ecfad80402()
63 .SetShapeFn([](InferenceContext* c) { in __anona0ecfad80502()
71 .SetShapeFn([](InferenceContext* c) { in __anona0ecfad80602()
75 Status RFFTShape(InferenceContext* c, const bool forward, const int rank) { in RFFTShape()
117 .SetShapeFn([](InferenceContext* c) { return RFFTShape(c, true, 1); }); in __anona0ecfad80702()
125 .SetShapeFn([](InferenceContext* c) { return RFFTShape(c, false, 1); }); in __anona0ecfad80802()
[all …]
|
| D | cudnn_rnn_ops.cc | 37 using shape_inference::InferenceContext;
54 .SetShapeFn([](InferenceContext* c) { in __anon815c5c9f0202()
83 .SetShapeFn([](InferenceContext* c) { in __anon815c5c9f0302()
126 .SetShapeFn([](InferenceContext* c) { in __anon815c5c9f0402()
173 .SetShapeFn([](InferenceContext* c) { in __anon815c5c9f0502()
224 .SetShapeFn([](InferenceContext* c) { in __anon815c5c9f0602()
261 .SetShapeFn([](InferenceContext* c) { in __anon815c5c9f0702()
301 .SetShapeFn([](InferenceContext* c) { in __anon815c5c9f0802()
328 .SetShapeFn([](InferenceContext* c) { in __anon815c5c9f0902()
335 c->set_output(i, c->Matrix(InferenceContext::kUnknownDim, in __anon815c5c9f0902()
[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 __anon18b155f40202()
90 Status ValidateTableType(InferenceContext* c, in ValidateTableType()
116 Status ValidateTableResourceHandle(InferenceContext* c, ShapeHandle keys, in ValidateTableResourceHandle()
168 .SetShapeFn([](InferenceContext* c) { in __anon18b155f40302()
191 .SetShapeFn([](InferenceContext* c) { in __anon18b155f40402()
[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 __anon52e493120202()
93 .SetShapeFn([](InferenceContext* c) { in __anon52e493120302()
137 .SetShapeFn([](InferenceContext* c) { in __anon52e493120402()
150 .SetShapeFn([](InferenceContext* c) { in __anon52e493120502()
173 .SetShapeFn([](InferenceContext* c) { in __anon52e493120602()
199 .SetShapeFn([](InferenceContext* c) { in __anon52e493120702()
[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 DecodeImageV2ShapeFn(InferenceContext* c) { in DecodeImageV2ShapeFn()
118 c->MakeShape({InferenceContext::kUnknownDim, in DecodeImageV2ShapeFn()
119 InferenceContext::kUnknownDim, channels_dim})); in DecodeImageV2ShapeFn()
124 Status EncodeImageShapeFn(InferenceContext* c) { in EncodeImageShapeFn()
[all …]
|
| D | string_ops.cc | 31 class InferenceContext;
35 using shape_inference::InferenceContext;
44 .SetShapeFn([](InferenceContext* c) { in __anon4354d5f10102()
64 .SetShapeFn([](InferenceContext* c) { in __anon4354d5f10202()
134 .SetShapeFn([](InferenceContext* c) { in __anon4354d5f10302()
158 .SetShapeFn([](InferenceContext* c) { in __anon4354d5f10402()
190 .SetShapeFn([](InferenceContext* c) { in __anon4354d5f10502()
195 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, 2)); in __anon4354d5f10502()
196 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in __anon4354d5f10502()
208 .SetShapeFn([](InferenceContext* c) { in __anon4354d5f10602()
[all …]
|
| D | dataset_ops.cc | 108 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon42b7b0c00102()
120 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon42b7b0c00202()
132 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon42b7b0c00302()
188 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon42b7b0c00402()
291 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon42b7b0c00502()
307 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon42b7b0c00602()
322 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon42b7b0c00702()
339 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon42b7b0c00802()
358 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon42b7b0c00902()
380 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon42b7b0c00a02()
[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 __anonea01102b0202()
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 __anonea01102b0502()
338 .SetShapeFn([](InferenceContext* c) { in __anonea01102b0602()
341 n_shape = c->Vector(InferenceContext::kUnknownDim); in __anonea01102b0602()
366 .SetShapeFn([](InferenceContext* c) { in __anonea01102b0702()
[all …]
|
| D | ragged_conversion_ops.cc | 24 using shape_inference::InferenceContext;
30 InferenceContext* c) { in ValidateRowPartitionTypesAndShapes()
92 Status RaggedTensorToSparseShapeFn(InferenceContext* c);
93 Status RaggedTensorToVariantShapeFn(InferenceContext* c);
94 Status RaggedTensorFromVariantShapeFn(InferenceContext* c);
95 Status RaggedTensorToVariantGradientShapeFn(InferenceContext* c);
96 Status RaggedTensorToTensorShapeFn(InferenceContext* c);
159 Status RaggedTensorToSparseShapeFn(InferenceContext* c) { in RaggedTensorToSparseShapeFn()
188 Status RaggedTensorToVariantShapeFn(InferenceContext* c) { in RaggedTensorToVariantShapeFn()
214 Status RaggedTensorToVariantGradientShapeFn(InferenceContext* c) { in RaggedTensorToVariantGradientShapeFn()
[all …]
|
| D | map_ops.cc | 26 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon3f27e5980202()
42 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon3f27e5980302()
54 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon3f27e5980402()
65 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon3f27e5980502()
81 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon3f27e5980602()
|
| D | resource_variable_ops.cc | 24 using ::tensorflow::shape_inference::InferenceContext;
32 Status ReadVariableShapeFn(InferenceContext* c) { in ReadVariableShapeFn()
46 Status ReadVariablesShapeFn(InferenceContext* c) { in ReadVariablesShapeFn()
86 .SetShapeFn([](InferenceContext* c) { in __anon50d007a70202()
108 .SetShapeFn([](InferenceContext* c) { in __anon50d007a70302()
169 Status CreateAssignShapeFn(InferenceContext* c) { in CreateAssignShapeFn()
218 Status VariableShapeShapeFn(InferenceContext* c) { in VariableShapeShapeFn()
226 : InferenceContext::kUnknownDim; in VariableShapeShapeFn()
245 .SetShapeFn([](InferenceContext* c) { in __anon50d007a70402()
306 Status ResourceScatterUpdateShape(InferenceContext* c) { in ResourceScatterUpdateShape()
[all …]
|
| D | experimental_dataset_ops.cc | 26 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon4afff42d0102()
39 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon4afff42d0202()
52 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon4afff42d0302()
86 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon4afff42d0402()
98 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon4afff42d0502()
169 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon4afff42d0602()
212 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon4afff42d0702()
257 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon4afff42d0802()
319 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon4afff42d0902()
335 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon4afff42d0a02()
[all …]
|
| D | array_ops.cc | 35 using shape_inference::InferenceContext;
41 Status GetAxisForPackAndUnpack(InferenceContext* c, int32 rank_after_pack, in GetAxisForPackAndUnpack()
64 Status PadKnown(InferenceContext* c, ShapeHandle input, in PadKnown()
81 Status PadShapeFn(InferenceContext* c) { in PadShapeFn()
121 Status TransposeShapeFn(InferenceContext* c) { in TransposeShapeFn()
187 Status SetOutputShapeForReshape(InferenceContext* c) { in SetOutputShapeForReshape()
290 .SetShapeFn([](InferenceContext* c) { in __anon42d741190202()
329 .SetShapeFn([](InferenceContext* c) { in __anon42d741190302()
406 .SetShapeFn([](InferenceContext* c) { in __anon42d741190402()
420 .SetShapeFn([](InferenceContext* c) { in __anon42d741190502()
[all …]
|
| D | stateful_random_ops.cc | 23 Status StatefulRandomShape(shape_inference::InferenceContext* c) { in StatefulRandomShape()
59 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon0d38c5e20102()
87 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon0d38c5e20202()
99 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon0d38c5e20302()
113 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon0d38c5e20402()
131 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon0d38c5e20502()
150 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon0d38c5e20602()
|
| D | nn_ops.cc | 30 using shape_inference::InferenceContext;
35 Status FractionalPoolShapeFn(InferenceContext* c) { in FractionalPoolShapeFn()
106 .SetShapeFn([](InferenceContext* c) { in __anonad10f3360202()
138 .SetShapeFn([](InferenceContext* c) { in __anonad10f3360302()
373 .SetShapeFn([](InferenceContext* c) { in __anonad10f3360402()
425 Status CommonFusedConvCalculations(InferenceContext* c, bool has_resize) { in CommonFusedConvCalculations()
545 .SetShapeFn([](InferenceContext* c) { in __anonad10f3360602()
558 .SetShapeFn([](InferenceContext* c) { in __anonad10f3360702()
587 .SetShapeFn([](InferenceContext* c) { in __anonad10f3360802()
606 .SetShapeFn([](InferenceContext* c) { in __anonad10f3360902()
[all …]
|
| D | list_ops.cc | 27 shape_inference::InferenceContext* c, in VerifyHandleData()
63 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonabedecb60202()
81 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonabedecb60302()
118 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonabedecb60402()
171 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonabedecb60502()
208 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonabedecb60602()
252 shape_inference::InferenceContext* c, in TensorListConcatShapeInference()
293 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonabedecb60702()
310 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonabedecb60802()
324 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonabedecb60902()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.h | 38 std::unique_ptr<shape_inference::InferenceContext> ic, const Node* node) in ExtendedInferenceContext()
53 shape_inference::InferenceContext* get_context() { in get_context()
60 std::unique_ptr<shape_inference::InferenceContext> inference_context_;
113 shape_inference::InferenceContext* GetContext(const Node* node) const { in GetContext()
162 static bool SameDefinedShape(shape_inference::InferenceContext* c,
169 shape_inference::InferenceContext* c,
191 const Node* node, shape_inference::InferenceContext* outer_context);
210 shape_inference::InferenceContext* outer_context);
223 shape_inference::InferenceContext* outer_context);
236 shape_inference::InferenceContext* outer_context);
[all …]
|
| /external/tensorflow/tensorflow/c/ |
| D | ops.cc | 32 using ::tensorflow::shape_inference::InferenceContext;
93 [shape_inference_func](InferenceContext* ctx) -> tensorflow::Status { in TF_OpDefinitionBuilderSetShapeInferenceFunction()
109 *handle = reinterpret_cast<InferenceContext*>(ctx)->Scalar(); in TF_ShapeInferenceContextScalar()
116 *handle = reinterpret_cast<InferenceContext*>(ctx)->Vector(size); in TF_ShapeInferenceContextVectorFromSize()
125 auto* cc_ctx = reinterpret_cast<InferenceContext*>(ctx); in TF_ShapeInferenceContextConcatenateShapes()
137 auto* cc_ctx = reinterpret_cast<InferenceContext*>(ctx); in TF_ShapeInferenceContextNumInputs()
145 auto* cc_ctx = reinterpret_cast<InferenceContext*>(ctx); in TF_ShapeInferenceContextGetInput()
157 auto* cc_ctx = reinterpret_cast<InferenceContext*>(ctx); in TF_ShapeInferenceContextRankKnown()
165 auto* cc_ctx = reinterpret_cast<InferenceContext*>(ctx); in TF_ShapeInferenceContextSetOutput()
196 auto* cc_ctx = reinterpret_cast<InferenceContext*>(ctx); \
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | serialization.h | 29 class InferenceContext; variable
31 flatbuffers::Offset<data::InferenceContext> Encode(
32 const InferenceContext& inference, flatbuffers::FlatBufferBuilder* builder);
35 const data::InferenceContext* fb_inference,
36 InferenceContext* inference);
|
| D | serialization_generated.h | 39 struct InferenceContext;
237 struct InferenceContext FLATBUFFERS_FINAL_CLASS : private flatbuffers::Table {
333 typedef InferenceContext Table;
337 fbb_.AddElement<uint8_t>(InferenceContext::VT_NEED_FLUSH, static_cast<uint8_t>(need_flush), 0); in add_need_flush()
340 …fbb_.AddElement<uint8_t>(InferenceContext::VT_FLUSH_PERIODICALLY, static_cast<uint8_t>(flush_perio… in add_flush_periodically()
343 fbb_.AddElement<int32_t>(InferenceContext::VT_FLUSH_PERIOD, flush_period, 0); in add_flush_period()
346 …fbb_.AddElement<uint8_t>(InferenceContext::VT_NEED_MANUAL_RELEASE, static_cast<uint8_t>(need_manua… in add_need_manual_release()
349 fbb_.AddElement<int8_t>(InferenceContext::VT_PRECISION, static_cast<int8_t>(precision), 0); in add_precision()
352 … fbb_.AddElement<int8_t>(InferenceContext::VT_STORAGE_TYPE, static_cast<int8_t>(storage_type), 0); in add_storage_type()
355 fbb_.AddOffset(InferenceContext::VT_NODES, nodes); in add_nodes()
[all …]
|