| /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;
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()
137 Status InferenceContext::input(StringPiece input_name, in input()
151 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 __anon5932d8840202()
48 .SetShapeFn([](InferenceContext* c) { return (*global_fn_ptr)(c); }); in __anon5932d8840302()
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 …]
|
| D | shape_inference_test.cc | 66 void Relax(InferenceContext* c, DimensionHandle d0, DimensionHandle d1, in Relax()
70 void Relax(InferenceContext* c, ShapeHandle s0, ShapeHandle s1, in Relax()
88 InferenceContext c(kVersion, def, op_def, {S({1, 5}), S({2, 5}), S({1, 3})}, in TEST_F()
125 InferenceContext c(kVersion, def, MakeOpDef(1, 1), {Unknown()}, {}, {}, {}); in TEST_F()
126 EXPECT_EQ(InferenceContext::kUnknownDim, in TEST_F()
127 c.Value(InferenceContext::kUnknownDim)); in TEST_F()
140 InferenceContext c(kVersion, def, MakeOpDef(1, 2), {S({1})}, {}, {}, {}); in TEST_F()
144 auto fn = [](InferenceContext* c) { in TEST_F()
155 auto fn = [](InferenceContext* c) { in TEST_F()
177 InferenceContext c(kVersion, def, MakeOpDef(1, 2), {S({1, 2, 3})}, {}, {}, in TEST_F()
[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 __anon5983aeac0202()
75 .SetShapeFn([](InferenceContext* c) { in __anon5983aeac0302()
79 0, c->Matrix(InferenceContext::kUnknownDim, c->Dim(a_shape, 0))); in __anon5983aeac0302()
80 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in __anon5983aeac0302()
95 .SetShapeFn([](InferenceContext* c) { in __anon5983aeac0402()
[all …]
|
| D | spectral_ops.cc | 24 using shape_inference::InferenceContext;
31 .SetShapeFn([](InferenceContext* c) { in __anone53cc5fc0102()
39 .SetShapeFn([](InferenceContext* c) { in __anone53cc5fc0202()
47 .SetShapeFn([](InferenceContext* c) { in __anone53cc5fc0302()
55 .SetShapeFn([](InferenceContext* c) { in __anone53cc5fc0402()
63 .SetShapeFn([](InferenceContext* c) { in __anone53cc5fc0502()
71 .SetShapeFn([](InferenceContext* c) { in __anone53cc5fc0602()
75 Status RFFTShape(InferenceContext* c, const bool forward, const int rank) { in RFFTShape()
117 .SetShapeFn([](InferenceContext* c) { return RFFTShape(c, true, 1); }); in __anone53cc5fc0702()
125 .SetShapeFn([](InferenceContext* c) { return RFFTShape(c, false, 1); }); in __anone53cc5fc0802()
[all …]
|
| D | cudnn_rnn_ops.cc | 37 using shape_inference::InferenceContext;
54 .SetShapeFn([](InferenceContext* c) { in __anon4fa58c430202()
83 .SetShapeFn([](InferenceContext* c) { in __anon4fa58c430302()
126 .SetShapeFn([](InferenceContext* c) { in __anon4fa58c430402()
173 .SetShapeFn([](InferenceContext* c) { in __anon4fa58c430502()
224 .SetShapeFn([](InferenceContext* c) { in __anon4fa58c430602()
261 .SetShapeFn([](InferenceContext* c) { in __anon4fa58c430702()
301 .SetShapeFn([](InferenceContext* c) { in __anon4fa58c430802()
328 .SetShapeFn([](InferenceContext* c) { in __anon4fa58c430902()
335 c->set_output(i, c->Matrix(InferenceContext::kUnknownDim, in __anon4fa58c430902()
[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 __anon3b8c08180202()
90 Status ValidateTableResourceHandle(InferenceContext* c, ShapeHandle keys, in ValidateTableResourceHandle()
165 .SetShapeFn([](InferenceContext* c) { in __anon3b8c08180302()
193 .SetShapeFn([](InferenceContext* c) { in __anon3b8c08180402()
209 .SetShapeFn([](InferenceContext* c) { in __anon3b8c08180502()
[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 __anon2d93d3360202()
93 .SetShapeFn([](InferenceContext* c) { in __anon2d93d3360302()
137 .SetShapeFn([](InferenceContext* c) { in __anon2d93d3360402()
150 .SetShapeFn([](InferenceContext* c) { in __anon2d93d3360502()
173 .SetShapeFn([](InferenceContext* c) { in __anon2d93d3360602()
199 .SetShapeFn([](InferenceContext* c) { in __anon2d93d3360702()
[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 SoftNMSShapeFn(InferenceContext* c) { in SoftNMSShapeFn()
[all …]
|
| D | string_ops.cc | 31 class InferenceContext;
35 using shape_inference::InferenceContext;
44 .SetShapeFn([](InferenceContext* c) { in __anon662f88150102()
64 .SetShapeFn([](InferenceContext* c) { in __anon662f88150202()
134 .SetShapeFn([](InferenceContext* c) { in __anon662f88150302()
158 .SetShapeFn([](InferenceContext* c) { in __anon662f88150402()
190 .SetShapeFn([](InferenceContext* c) { in __anon662f88150502()
195 c->set_output(0, c->Matrix(InferenceContext::kUnknownDim, 2)); in __anon662f88150502()
196 c->set_output(1, c->Vector(InferenceContext::kUnknownDim)); in __anon662f88150502()
208 .SetShapeFn([](InferenceContext* c) { in __anon662f88150602()
[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 __anonb84a3fcf0202()
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 __anonb84a3fcf0502()
338 .SetShapeFn([](InferenceContext* c) { in __anonb84a3fcf0602()
341 n_shape = c->Vector(InferenceContext::kUnknownDim); in __anonb84a3fcf0602()
366 .SetShapeFn([](InferenceContext* c) { in __anonb84a3fcf0702()
[all …]
|
| D | dataset_ops.cc | 108 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonc0e8a7640102()
120 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonc0e8a7640202()
132 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonc0e8a7640302()
172 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonc0e8a7640402()
257 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonc0e8a7640502()
273 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonc0e8a7640602()
288 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonc0e8a7640702()
305 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonc0e8a7640802()
327 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonc0e8a7640902()
345 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonc0e8a7640a02()
[all …]
|
| D | experimental_dataset_ops.cc | 26 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon03b595510102()
39 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon03b595510202()
72 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon03b595510302()
84 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon03b595510402()
137 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon03b595510502()
179 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon03b595510602()
246 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon03b595510702()
262 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon03b595510802()
393 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon03b595510902()
405 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon03b595510a02()
[all …]
|
| 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()
85 .SetShapeFn([](InferenceContext* c) { in __anond0909b4b0202()
107 .SetShapeFn([](InferenceContext* c) { in __anond0909b4b0302()
168 Status CreateAssignShapeFn(InferenceContext* c) { in CreateAssignShapeFn()
217 Status VariableShapeShapeFn(InferenceContext* c) { in VariableShapeShapeFn()
225 : InferenceContext::kUnknownDim; in VariableShapeShapeFn()
244 .SetShapeFn([](InferenceContext* c) { in __anond0909b4b0402()
305 Status ResourceScatterUpdateShape(InferenceContext* c) { in ResourceScatterUpdateShape()
[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 tensorflow::Status RaggedTensorToTensorShapeFn(InferenceContext* c);
149 Status RaggedTensorToSparseShapeFn(InferenceContext* c) { in RaggedTensorToSparseShapeFn()
178 Status RaggedTensorToVariantShapeFn(InferenceContext* c) { in RaggedTensorToVariantShapeFn()
204 Status RaggedTensorFromVariantShapeFn(InferenceContext* c) { in RaggedTensorFromVariantShapeFn()
224 tensorflow::Status RaggedTensorToTensorShapeFn(InferenceContext* c) { in RaggedTensorToTensorShapeFn()
|
| D | nn_ops.cc | 28 using shape_inference::InferenceContext;
33 Status FractionalPoolShapeFn(InferenceContext* c) { in FractionalPoolShapeFn()
89 .SetShapeFn([](InferenceContext* c) { in __anon87c0335a0202()
110 .SetShapeFn([](InferenceContext* c) { in __anon87c0335a0302()
142 .SetShapeFn([](InferenceContext* c) { in __anon87c0335a0402()
354 .SetShapeFn([](InferenceContext* c) { in __anon87c0335a0502()
377 .SetShapeFn([](InferenceContext* c) { in __anon87c0335a0602()
410 Status CommonFusedConvCalculations(InferenceContext* c, bool has_resize) { in CommonFusedConvCalculations()
530 .SetShapeFn([](InferenceContext* c) { in __anon87c0335a0802()
543 .SetShapeFn([](InferenceContext* c) { in __anon87c0335a0902()
[all …]
|
| D | list_ops.cc | 26 shape_inference::InferenceContext* c, in VerifyHandleData()
57 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonef6dc5da0202()
75 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonef6dc5da0302()
112 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonef6dc5da0402()
165 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonef6dc5da0502()
202 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonef6dc5da0602()
246 shape_inference::InferenceContext* c, in TensorListConcatShapeInference()
287 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonef6dc5da0702()
304 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonef6dc5da0802()
318 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonef6dc5da0902()
[all …]
|
| D | array_ops.cc | 34 using shape_inference::InferenceContext;
40 Status GetAxisForPackAndUnpack(InferenceContext* c, int32 rank_after_pack, in GetAxisForPackAndUnpack()
63 Status PadKnown(InferenceContext* c, ShapeHandle input, in PadKnown()
80 Status PadShapeFn(InferenceContext* c) { in PadShapeFn()
120 Status TransposeShapeFn(InferenceContext* c) { in TransposeShapeFn()
186 Status SetOutputShapeForReshape(InferenceContext* c) { in SetOutputShapeForReshape()
290 .SetShapeFn([](InferenceContext* c) { in __anonf6523ebd0202()
329 .SetShapeFn([](InferenceContext* c) { in __anonf6523ebd0302()
406 .SetShapeFn([](InferenceContext* c) { in __anonf6523ebd0402()
420 .SetShapeFn([](InferenceContext* c) { in __anonf6523ebd0502()
[all …]
|
| D | parsing_ops.cc | 25 using shape_inference::InferenceContext;
33 const ShapeHandle& prefix, InferenceContext* c, in AddDenseOutputShapes()
46 int64 rank_delta, InferenceContext* c, in AddSparseOutputShapes()
67 InferenceContext* c, int* output_idx) { in AddRaggedOutputShapes()
89 InferenceContext* c, int* output_idx) { in AddDenseLengthsShapes()
105 .SetShapeFn([](InferenceContext* c) { in __anon7727dd120202()
109 c->input(0), c->Vector(InferenceContext::kUnknownDim), &out)); in __anon7727dd120202()
120 .SetShapeFn([](InferenceContext* c) { in __anon7727dd120302()
160 .SetShapeFn([](InferenceContext* c) { in __anon7727dd120402()
204 .SetShapeFn([](InferenceContext* c) { in __anon7727dd120502()
[all …]
|
| D | stateful_random_ops.cc | 22 Status StatefulRandomShape(shape_inference::InferenceContext* c) { in StatefulRandomShape()
58 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonefb820860102()
76 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonefb820860202()
89 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonefb820860302()
107 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonefb820860402()
126 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anonefb820860502()
|
| /external/tensorflow/tensorflow/c/ |
| D | ops.cc | 32 using ::tensorflow::shape_inference::InferenceContext;
93 [shape_inference_func](InferenceContext* ctx) -> tensorflow::Status { in TF_OpDefinitionBuilderSetShapeInferenceFunction()
110 *handle = reinterpret_cast<InferenceContext*>(ctx)->Vector(size); in TF_ShapeInferenceContextVectorFromSize()
119 auto* cc_ctx = reinterpret_cast<InferenceContext*>(ctx); in TF_ShapeInferenceContextConcatenateShapes()
131 auto* cc_ctx = reinterpret_cast<InferenceContext*>(ctx); in TF_ShapeInferenceContextNumInputs()
139 auto* cc_ctx = reinterpret_cast<InferenceContext*>(ctx); in TF_ShapeInferenceContextGetInput()
151 auto* cc_ctx = reinterpret_cast<InferenceContext*>(ctx); in TF_ShapeInferenceContextRankKnown()
159 auto* cc_ctx = reinterpret_cast<InferenceContext*>(ctx); in TF_ShapeInferenceContextSetOutput()
190 auto* cc_ctx = reinterpret_cast<InferenceContext*>(ctx); \
204 auto* cc_ctx = reinterpret_cast<InferenceContext*>(ctx); \ in DEFINE_TF_GETATTR()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.h | 37 std::unique_ptr<shape_inference::InferenceContext> ic, const Node* node) in ExtendedInferenceContext()
52 shape_inference::InferenceContext* get_context() { in get_context()
59 std::unique_ptr<shape_inference::InferenceContext> inference_context_;
112 shape_inference::InferenceContext* GetContext(const Node* node) const { in GetContext()
161 static bool SameDefinedShape(shape_inference::InferenceContext* c,
168 shape_inference::InferenceContext* c,
219 Status ConstantPartialShape(shape_inference::InferenceContext* target_context,
225 shape_inference::InferenceContext* ctx,
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 46 using shape_inference::InferenceContext;
87 if (InferenceContext::RankKnown(*result)) { in Merge()
92 if (InferenceContext::RankKnown(h1)) { in Merge()
106 if (!InferenceContext::ValueKnown(d)) { in ExtractValue()
110 int64 val = InferenceContext::Value(d); in ExtractValue()
126 const int64 dim1 = InferenceContext::Value(d1); in Merge()
127 const int64 dim2 = InferenceContext::Value(d2); in Merge()
357 InferenceContext* ic, const ShapeHandle& shape, in IsShapeFullyDefinedIntegerVectorOrScalar()
369 TensorProto MakeTensorProtoFromShape(InferenceContext* ic, in MakeTensorProtoFromShape()
392 NodeDef MakeConstNodeDefFromTensorProto(InferenceContext* ic, in MakeConstNodeDefFromTensorProto()
[all …]
|