| /external/tensorflow/tensorflow/python/framework/ |
| D | test_ops.cc | 32 .SetShapeFn(shape_inference::ScalarShape);
37 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon9d3b0e9e0102()
38 shape_inference::ShapeHandle out; in __anon9d3b0e9e0102()
47 .SetShapeFn(shape_inference::ScalarShape);
52 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon9d3b0e9e0202()
56 return shape_inference::ScalarShape(c); in __anon9d3b0e9e0202()
60 .SetShapeFn(shape_inference::UnknownShape)
65 .SetShapeFn(shape_inference::UnknownShape);
69 .SetShapeFn(shape_inference::UnknownShape);
76 .SetShapeFn(shape_inference::UnchangedShape);
[all …]
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | experimental_dataset_ops.cc | 29 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anond6112e9c0102()
30 shape_inference::ShapeHandle unused; in __anond6112e9c0102()
33 return shape_inference::ScalarShape(c); in __anond6112e9c0102()
44 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anond6112e9c0202()
45 shape_inference::ShapeHandle unused; in __anond6112e9c0202()
48 return shape_inference::ScalarShape(c); in __anond6112e9c0202()
59 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anond6112e9c0302()
60 shape_inference::ShapeHandle unused; in __anond6112e9c0302()
63 return shape_inference::ScalarShape(c); in __anond6112e9c0302()
74 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anond6112e9c0402()
[all …]
|
| D | dataset_ops.cc | 48 .SetShapeFn(shape_inference::ScalarShape);
65 .SetShapeFn(shape_inference::ScalarShape);
75 .SetShapeFn(shape_inference::ScalarShape);
94 .SetShapeFn(shape_inference::ScalarShape);
105 .SetShapeFn(shape_inference::ScalarShape);
116 .SetShapeFn(shape_inference::ScalarShape);
127 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon8d7b4acf0102()
128 shape_inference::ShapeHandle count_shape; in __anon8d7b4acf0102()
130 return shape_inference::ScalarShape(c); in __anon8d7b4acf0102()
142 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon8d7b4acf0202()
[all …]
|
| D | risc_ops.cc | 25 Status RiscBinaryNonBroadcastOpShapeFn(shape_inference::InferenceContext* c) { in RiscBinaryNonBroadcastOpShapeFn()
52 .SetShapeFn(shape_inference::UnchangedShape);
70 .SetShapeFn(shape_inference::UnchangedShape);
86 .SetShapeFn(shape_inference::UnknownShape);
95 .SetShapeFn(shape_inference::UnknownShape);
102 .SetShapeFn(shape_inference::UnchangedShape);
108 .SetShapeFn(shape_inference::UnchangedShape);
115 .SetShapeFn(shape_inference::UnknownShape);
124 .SetShapeFn(shape_inference::ConcatV2Shape);
136 .SetShapeFn(shape_inference::UnknownShape);
[all …]
|
| D | list_ops.cc | 28 shape_inference::InferenceContext* c, in VerifyHandleData()
29 const std::vector<shape_inference::ShapeAndType>& shapes_and_types, in VerifyHandleData()
37 const shape_inference::ShapeAndType& list_shape_type = shapes_and_types[0]; in VerifyHandleData()
48 const std::vector<shape_inference::ShapeAndType>* handle_data) { in IsValidTensorListHandleData()
53 shape_inference::ShapeHandle GetElementShapeFromHandleData( in GetElementShapeFromHandleData()
54 const std::vector<shape_inference::ShapeAndType>& shapes_and_types) { in GetElementShapeFromHandleData()
66 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon398e29250202()
70 shape_inference::ShapeHandle element_shape; in __anon398e29250202()
75 0, std::vector<shape_inference::ShapeAndType>{ in __anon398e29250202()
87 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon398e29250302()
[all …]
|
| D | collective_ops.cc | 37 .SetShapeFn(shape_inference::UnchangedShape);
51 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon9f208cb30102()
53 shape_inference::ShapeHandle unused; in __anon9f208cb30102()
56 shape_inference::ShapeHandle in_subshape; in __anon9f208cb30102()
63 shape_inference::ShapeHandle output_first_dim_as_shape; in __anon9f208cb30102()
65 shape_inference::InferenceContext::kUnknownDim) { in __anon9f208cb30102()
67 c->Vector(shape_inference::InferenceContext::kUnknownDim); in __anon9f208cb30102()
71 std::vector<shape_inference::DimensionHandle> output_first_dim; in __anon9f208cb30102()
77 shape_inference::ShapeHandle out; in __anon9f208cb30102()
96 .SetShapeFn(shape_inference::ExplicitShape);
[all …]
|
| D | summary_ops.cc | 25 .SetShapeFn(shape_inference::ScalarShape);
33 .SetShapeFn(shape_inference::NoOutputs);
41 .SetShapeFn(shape_inference::NoOutputs);
45 .SetShapeFn(shape_inference::NoOutputs);
49 .SetShapeFn(shape_inference::NoOutputs);
58 .SetShapeFn(shape_inference::NoOutputs);
64 .SetShapeFn(shape_inference::NoOutputs);
69 .SetShapeFn(shape_inference::NoOutputs);
77 .SetShapeFn(shape_inference::NoOutputs);
85 .SetShapeFn(shape_inference::NoOutputs);
[all …]
|
| D | boosted_trees_ops.cc | 27 using shape_inference::DimensionHandle;
28 using shape_inference::InferenceContext;
29 using shape_inference::ShapeHandle;
36 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon9348109b0102()
37 shape_inference::ShapeHandle unused_input; in __anon9348109b0102()
57 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon9348109b0202()
63 shape_inference::ShapeHandle node_id_range_shape; in __anon9348109b0202()
64 shape_inference::ShapeHandle unused_shape; in __anon9348109b0202()
69 shape_inference::ShapeHandle summary_shape_base; in __anon9348109b0202()
75 shape_inference::ShapeHandle summary_shape; in __anon9348109b0202()
[all …]
|
| D | math_ops.cc | 23 using shape_inference::DimensionHandle;
24 using shape_inference::InferenceContext;
25 using shape_inference::ShapeHandle;
52 std::vector<shape_inference::ShapeAndType> cur_shapes_and_types; in __anonfffccbd30102()
114 .SetShapeFn(shape_inference::ExplicitShape);
127 .SetShapeFn(shape_inference::BatchMatMulShape);
138 .SetShapeFn(shape_inference::BatchMatMulV2Shape);
155 .SetShapeFn(shape_inference::BatchMatMulV2Shape);
165 .SetShapeFn(shape_inference::BatchMatMulShape);
174 .SetShapeFn(shape_inference::BatchMatMulV2Shape);
[all …]
|
| D | spectral_ops.cc | 23 using shape_inference::DimensionHandle;
24 using shape_inference::InferenceContext;
25 using shape_inference::ShapeHandle;
32 return shape_inference::UnchangedShapeWithRankAtLeast(c, 1); in __anon4423d6c70102()
40 return shape_inference::UnchangedShapeWithRankAtLeast(c, 1); in __anon4423d6c70202()
48 return shape_inference::UnchangedShapeWithRankAtLeast(c, 2); in __anon4423d6c70302()
56 return shape_inference::UnchangedShapeWithRankAtLeast(c, 2); in __anon4423d6c70402()
64 return shape_inference::UnchangedShapeWithRankAtLeast(c, 3); in __anon4423d6c70502()
72 return shape_inference::UnchangedShapeWithRankAtLeast(c, 3); in __anon4423d6c70602()
163 .SetShapeFn(shape_inference::UnknownShape)
[all …]
|
| D | data_flow_ops.cc | 23 using shape_inference::DimensionHandle;
24 using shape_inference::InferenceContext;
25 using shape_inference::ShapeHandle;
40 return shape_inference::UnknownShape(c); in DequeueManyV2Shape()
62 return shape_inference::UnknownShape(c); in __anonf41369fa0202()
204 .SetShapeFn(shape_inference::ScalarShape);
224 .SetShapeFn(shape_inference::ScalarShape);
244 .SetShapeFn(shape_inference::ScalarShape);
264 .SetShapeFn(shape_inference::ScalarShape);
277 .SetShapeFn(shape_inference::UnknownShape);
[all …]
|
| D | stateful_random_ops.cc | 23 Status StatefulRandomShape(shape_inference::InferenceContext* c) { in StatefulRandomShape()
24 using shape_inference::ShapeHandle; in StatefulRandomShape()
59 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon1a40fef10102()
60 using shape_inference::ShapeHandle; in __anon1a40fef10102()
87 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon1a40fef10202()
88 shape_inference::ShapeHandle unused; in __anon1a40fef10202()
99 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon1a40fef10302()
100 shape_inference::ShapeHandle unused; in __anon1a40fef10302()
113 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon1a40fef10402()
114 using shape_inference::ShapeHandle; in __anon1a40fef10402()
[all …]
|
| D | functional_ops.cc | 22 using shape_inference::InferenceContext;
41 const std::vector<shape_inference::ShapeAndType>* handle_type = in __anonff7c821c0102()
63 .SetShapeFn(shape_inference::UnknownShape);
76 .SetShapeFn(shape_inference::UnknownShape)
93 Status IfShapeInferenceFn(shape_inference::InferenceContext* c) { in IfShapeInferenceFn()
98 if (output_shapes.empty()) return shape_inference::UnknownShape(c); in IfShapeInferenceFn()
105 shape_inference::ShapeHandle output_shape_handle; in IfShapeInferenceFn()
138 Status CaseShapeInferenceFn(shape_inference::InferenceContext* c) { in CaseShapeInferenceFn()
143 if (output_shapes.empty()) return shape_inference::UnknownShape(c); in CaseShapeInferenceFn()
150 shape_inference::ShapeHandle output_shape_handle; in CaseShapeInferenceFn()
[all …]
|
| D | logging_ops.cc | 23 using shape_inference::InferenceContext;
31 .SetShapeFn(shape_inference::NoOutputs);
45 .SetShapeFn(shape_inference::UnchangedShape);
79 .SetShapeFn(shape_inference::ScalarShape);
88 .SetShapeFn(shape_inference::ScalarShape);
100 .SetShapeFn(shape_inference::ScalarShape);
108 .SetShapeFn(shape_inference::ScalarShape);
116 .SetShapeFn(shape_inference::ScalarShape)
122 .SetShapeFn(shape_inference::ScalarShape);
|
| D | tpu_embedding_ops.cc | 65 .SetShapeFn([](shape_inference::InferenceContext* c) -> Status { in __anon3e675d600102()
78 shape_inference::ShapeHandle output_shape; in __anon3e675d600102()
92 .SetShapeFn([](shape_inference::InferenceContext *c) { in __anon3e675d600202()
104 .SetShapeFn([](shape_inference::InferenceContext* c) -> Status { in __anon3e675d600302()
107 std::vector<shape_inference::ShapeHandle> learning_rates; in __anon3e675d600302()
111 shape_inference::ShapeHandle learning_rates_shape; in __anon3e675d600302()
125 .SetShapeFn(shape_inference::UnknownShape);
139 .SetShapeFn([](shape_inference::InferenceContext* c) -> Status { in __anon3e675d600402()
168 .SetShapeFn(shape_inference::UnknownShape);
185 .SetShapeFn(shape_inference::UnknownShape);
[all …]
|
| D | string_ops.cc | 30 namespace shape_inference { namespace
34 using shape_inference::DimensionHandle;
35 using shape_inference::InferenceContext;
36 using shape_inference::ShapeHandle;
58 .SetShapeFn(shape_inference::UnchangedShape);
75 .SetShapeFn(shape_inference::UnchangedShape);
81 .SetShapeFn(shape_inference::UnchangedShape);
88 .SetShapeFn(shape_inference::UnchangedShape)
103 .SetShapeFn(shape_inference::UnchangedShape);
109 .SetShapeFn(shape_inference::UnchangedShape);
[all …]
|
| D | uniform_quant_ops.cc | 24 using shape_inference::DimensionHandle;
25 using shape_inference::ShapeHandle;
28 Status ScalesZeroPointsShapeValid(shape_inference::InferenceContext* context, in ScalesZeroPointsShapeValid()
31 const int32_t scales_rank = shape_inference::InferenceContext::Rank(scales); in ScalesZeroPointsShapeValid()
33 shape_inference::InferenceContext::Rank(zero_points); in ScalesZeroPointsShapeValid()
35 if (scales_rank == shape_inference::InferenceContext::kUnknownRank || in ScalesZeroPointsShapeValid()
36 zero_points_rank == shape_inference::InferenceContext::kUnknownRank) { in ScalesZeroPointsShapeValid()
57 Status DotHybridShape(shape_inference::InferenceContext* context) { in DotHybridShape()
97 .SetShapeFn(shape_inference::UnchangedShape);
114 .SetShapeFn(shape_inference::UnchangedShape);
[all …]
|
| D | nccl_ops.cc | 21 using shape_inference::InferenceContext;
22 using shape_inference::ShapeHandle;
32 .SetShapeFn(shape_inference::UnchangedShape);
43 .SetShapeFn(shape_inference::UnchangedShape);
52 .SetShapeFn(shape_inference::NoOutputs)
75 .SetShapeFn(shape_inference::UnchangedShape)
100 .SetShapeFn(shape_inference::UnchangedShape);
108 .SetShapeFn(shape_inference::NoOutputs)
|
| D | tpu_infeed_ops.cc | 22 using shape_inference::InferenceContext;
23 using shape_inference::ShapeHandle;
30 .SetShapeFn(shape_inference::ExplicitShape);
38 .SetShapeFn(shape_inference::NoOutputs)
47 .SetShapeFn(shape_inference::NoOutputs)
72 .SetShapeFn(shape_inference::ScalarShape);
80 .SetShapeFn(shape_inference::ScalarShape);
85 .SetShapeFn(shape_inference::NoOutputs);
|
| D | batch_ops.cc | 59 .SetShapeFn(shape_inference::UnknownShape)
77 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon1af940cb0102()
78 std::vector<shape_inference::ShapeHandle> in_shapes; in __anon1af940cb0102()
80 std::vector<shape_inference::ShapeHandle> out_shapes(in_shapes.size()); in __anon1af940cb0102()
89 {c->MakeShape({shape_inference::DimensionOrConstant(c->UnknownDim()), in __anon1af940cb0102()
90 shape_inference::DimensionOrConstant(3)})})); in __anon1af940cb0102()
104 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon1af940cb0202()
105 shape_inference::ShapeHandle out_shape; in __anon1af940cb0202()
121 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon1af940cb0302()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.h | 26 namespace shape_inference {
47 Status UnchangedShape(shape_inference::InferenceContext* c);
50 inline Status UnchangedShapeWithRank(shape_inference::InferenceContext* c, in UnchangedShapeWithRank()
60 shape_inference::InferenceContext* c, int32_t 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()
96 Status DatasetIteratorShape(shape_inference::InferenceContext* c);
104 shape_inference::InferenceContext* context);
107 Status MatMulShape(shape_inference::InferenceContext* c);
[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_;
100 shape_inference::ShapeHandle shape);
113 shape_inference::InferenceContext* GetContext(const Node* node) const { in GetContext()
162 static bool SameDefinedShape(shape_inference::InferenceContext* c,
163 shape_inference::ShapeHandle s0,
164 shape_inference::ShapeHandle s1);
169 shape_inference::InferenceContext* c,
170 const std::vector<shape_inference::ShapeAndType>& existing,
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| D | xla_ops.cc | 39 Status UnchangedRank(shape_inference::InferenceContext* c) { in UnchangedRank()
56 .SetShapeFn(shape_inference::UnknownShape)
78 .SetShapeFn(shape_inference::UnknownShape)
114 .SetShapeFn(shape_inference::UnknownShape)
205 static Status XlaDotShapeFunction(shape_inference::InferenceContext* c) { in XlaDotShapeFunction()
206 shape_inference::ShapeHandle lhs_shape_handle = c->input(0); in XlaDotShapeFunction()
207 shape_inference::ShapeHandle rhs_shape_handle = c->input(1); in XlaDotShapeFunction()
209 return shape_inference::UnknownShape(c); in XlaDotShapeFunction()
235 shape_inference::DimensionHandle unused; in XlaDotShapeFunction()
259 std::vector<shape_inference::DimensionHandle> output_dims; in XlaDotShapeFunction()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xrt/ops/ |
| D | xrt_state_ops.cc | 32 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
45 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
63 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
89 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
104 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
121 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
139 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
152 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
165 .SetShapeFn(tensorflow::shape_inference::ScalarShape)
180 .SetShapeFn(tensorflow::shape_inference::UnknownShape)
[all …]
|
| /external/tensorflow/tensorflow/core/tpu/ops/ |
| D | host_compute_ops.cc | 23 using shape_inference::InferenceContext;
24 using shape_inference::ShapeHandle;
34 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon7708b7eb0102()
35 return ::tensorflow::shape_inference::UnknownShape(c); in __anon7708b7eb0102()
66 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon7708b7eb0202()
79 shape_inference::ShapeHandle handle; in __anon7708b7eb0202()
88 return ::tensorflow::shape_inference::UnknownShape(c); in __anon7708b7eb0202()
96 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon7708b7eb0302()
97 return ::tensorflow::shape_inference::UnknownShape(c); in __anon7708b7eb0302()
107 .SetShapeFn([](shape_inference::InferenceContext* c) { in __anon7708b7eb0402()
[all …]
|