| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.h | 133 struct ShapeAndType { struct
134 ShapeAndType() {} in ShapeAndType() function
135 ShapeAndType(ShapeHandle s, DataType t) : shape(s), dtype(t) {} in ShapeAndType() function
169 std::vector<std::unique_ptr<std::vector<ShapeAndType>>>
349 string DebugString(const ShapeAndType& shape_and_type);
350 string DebugString(gtl::ArraySlice<ShapeAndType> shape_and_types);
545 const std::vector<ShapeAndType>& shapes_and_types) TF_MUST_USE_RESULT;
550 const std::vector<ShapeAndType>& shapes_and_types) TF_MUST_USE_RESULT;
563 const std::vector<ShapeAndType>& shapes_and_types) TF_MUST_USE_RESULT;
568 const std::vector<ShapeAndType>& shapes_and_types) TF_MUST_USE_RESULT;
[all …]
|
| D | shape_inference.cc | 63 std::vector<std::unique_ptr<std::vector<ShapeAndType>>> handle_data( in InferenceContext()
70 handle_data[i].reset(new std::vector<ShapeAndType>(v->size())); in InferenceContext()
90 std::vector<std::unique_ptr<std::vector<ShapeAndType>>> in InferenceContext()
193 std::vector<std::unique_ptr<std::vector<ShapeAndType>>> input_handle_data) { in PostInputInit()
292 string InferenceContext::DebugString(const ShapeAndType& shape_and_type) { in DebugString()
298 gtl::ArraySlice<ShapeAndType> shape_and_types) { in DebugString()
300 for (const ShapeAndType& s : shape_and_types) { in DebugString()
1136 const std::vector<ShapeAndType>& shapes_and_types, in MergeHandleShapesAndTypes()
1137 std::vector<ShapeAndType>* to_update) { in MergeHandleShapesAndTypes()
1141 std::vector<ShapeAndType> new_values(shapes_and_types.size()); in MergeHandleShapesAndTypes()
[all …]
|
| D | shape_inference_testutil.h | 34 typedef std::pair<string, DataType> ShapeAndType; typedef
39 std::vector<std::vector<ShapeAndType>*>
|
| D | shape_inference_testutil.cc | 45 std::vector<std::unique_ptr<std::vector<shape_inference::ShapeAndType>>> in InferShapes()
51 std::unique_ptr<std::vector<ShapeAndType>> v( in InferShapes()
52 new std::vector<ShapeAndType>()); in InferShapes()
|
| D | shape_inference_test.cc | 1622 [&](int idx, const std::vector<ShapeAndType>& shapes_and_types) { in TestMergeHandles()
1634 std::vector<ShapeAndType> t{{make_shape({1, 2, 3}), DT_FLOAT}, in TestMergeHandles()
1639 std::vector<ShapeAndType> v = *get_shapes_and_types_from_context(0); in TestMergeHandles()
1649 0, std::vector<ShapeAndType>{{make_shape({1, 2, 3}), DT_FLOAT}})); in TestMergeHandles()
1733 [&](int idx, const std::vector<ShapeAndType>& shapes_and_types) { in TestRelaxHandles()
1745 std::vector<ShapeAndType> t{{make_shape({1, 2, 3}), DT_FLOAT}, in TestRelaxHandles()
1750 std::vector<ShapeAndType> v = *get_shapes_and_types_from_context(0); in TestRelaxHandles()
1760 0, std::vector<ShapeAndType>{{make_shape({1, 2, 3}), DT_FLOAT}})); in TestRelaxHandles()
|
| D | common_shape_fns.h | 350 InferenceContext* c, std::vector<ShapeAndType>* shape_and_type);
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_csr_matrix_ops.cc | 26 using shape_inference::ShapeAndType;
30 ShapeAndType* shape_and_type) { in GetVariantInput()
83 {ShapeAndType{dense_shape, dtype}}); in __anond96a6a670102()
94 ShapeAndType sparse_matrix_shape_and_type; in __anond96a6a670202()
146 {ShapeAndType{dense_shape, dtype}}); in __anond96a6a670302()
156 ShapeAndType sparse_matrix_shape_and_type; in __anond96a6a670402()
175 ShapeAndType sparse_matrix_shape_and_type; in __anond96a6a670502()
203 ShapeAndType sparse_matrix_shape_and_type; in __anond96a6a670602()
233 ShapeAndType sparse_matrix_shape_and_type; in __anond96a6a670702()
304 ShapeAndType sparse_matrix_shape_and_type; in __anond96a6a670802()
[all …]
|
| D | sparse_csr_matrix_ops_test.cc | 45 std::vector<ShapeInferenceTestOp::ShapeAndType> shapes_and_types(1); in TEST()
72 std::vector<ShapeInferenceTestOp::ShapeAndType> shapes_and_types(1); in TEST()
84 std::vector<ShapeInferenceTestOp::ShapeAndType> shapes_and_types(1); in TEST()
104 std::vector<ShapeInferenceTestOp::ShapeAndType> a_shapes_and_types(1); in TEST()
183 std::vector<ShapeInferenceTestOp::ShapeAndType> a_shapes_and_types(1); in TEST()
184 std::vector<ShapeInferenceTestOp::ShapeAndType> b_shapes_and_types(1); in TEST()
221 std::vector<ShapeInferenceTestOp::ShapeAndType> a_shapes_and_types(1); in TEST()
222 std::vector<ShapeInferenceTestOp::ShapeAndType> b_shapes_and_types(1); in TEST()
291 std::vector<ShapeInferenceTestOp::ShapeAndType> shapes_and_types(1); in TEST()
306 std::vector<ShapeInferenceTestOp::ShapeAndType> shapes_and_types(1); in TEST()
[all …]
|
| D | list_ops.cc | 27 const std::vector<shape_inference::ShapeAndType>& shapes_and_types, in VerifyHandleData()
35 const shape_inference::ShapeAndType& list_shape_type = shapes_and_types[0]; in VerifyHandleData()
47 const std::vector<shape_inference::ShapeAndType>& shapes_and_types) { in GetElementShapeFromHandleData()
65 0, std::vector<shape_inference::ShapeAndType>{ in __anonef6dc5da0202()
87 const shape_inference::ShapeAndType& list_shape_type = in __anonef6dc5da0302()
102 0, std::vector<shape_inference::ShapeAndType>{ in __anonef6dc5da0302()
134 const shape_inference::ShapeAndType& list_shape_type = in __anonef6dc5da0402()
149 0, std::vector<shape_inference::ShapeAndType>{ in __anonef6dc5da0402()
175 const shape_inference::ShapeAndType& list_shape_type = in __anonef6dc5da0502()
212 const shape_inference::ShapeAndType& list_shape_type = in __anonef6dc5da0602()
[all …]
|
| D | resource_variable_ops.cc | 25 using ::tensorflow::shape_inference::ShapeAndType;
33 std::vector<ShapeAndType> shape_and_type; in ReadVariableShapeFn()
38 std::vector<ShapeAndType> variant_shape_and_type; in ReadVariableShapeFn()
56 ShapeAndType shape_and_type; in ReadVariablesShapeFn()
94 std::vector<ShapeAndType>{{s, t}}); in __anond0909b4b0202()
127 i, std::vector<ShapeAndType>{{s, dtypes[i]}}); in __anond0909b4b0302()
169 std::vector<ShapeAndType> handle_shape_and_type; in CreateAssignShapeFn()
245 std::vector<ShapeAndType> handle_shape_and_type; in __anond0909b4b0402()
286 std::vector<ShapeAndType> variant_shape_and_type; in __anond0909b4b0402()
306 std::vector<ShapeAndType> handle_shape_and_type; in ResourceScatterUpdateShape()
|
| D | lookup_ops.cc | 26 using shape_inference::ShapeAndType;
94 ShapeAndType* output_shape_and_type) { in ValidateTableResourceHandle()
100 const ShapeAndType& key_shape_and_type = (*handle_data)[0]; in ValidateTableResourceHandle()
101 const ShapeAndType& value_shape_and_type = (*handle_data)[1]; in ValidateTableResourceHandle()
173 ShapeAndType value_shape_and_type; in __anon3b8c08180302()
272 ShapeAndType value_shape_and_type; in __anon3b8c08180802()
331 0, std::vector<ShapeAndType>{{key_s, key_t}, {value, value_t}}); in MutableHashTableShape()
|
| D | data_flow_ops_test.cc | 205 std::vector<ShapeInferenceTestOp::ShapeAndType> shapes_and_types; in TEST()
229 std::vector<ShapeInferenceTestOp::ShapeAndType> shapes_and_types; in TEST()
279 std::vector<ShapeInferenceTestOp::ShapeAndType> shapes_and_types; in TEST()
|
| D | tpu_replication_ops.cc | 60 const std::vector<shape_inference::ShapeAndType>* shapes_and_types = in __anona12575700102()
|
| D | math_ops.cc | 52 std::vector<shape_inference::ShapeAndType> cur_shapes_and_types; in __anonb5dc68880102()
785 std::vector<shape_inference::ShapeAndType> merged_handle_data(size); in __anonb5dc68880402()
793 const shape_inference::ShapeAndType& s1 = (*handle_data_1)[i]; in __anonb5dc68880402()
794 const shape_inference::ShapeAndType& s2 = (*handle_data_2)[i]; in __anonb5dc68880402()
874 std::vector<shape_inference::ShapeAndType> merged_handle_data(size); in __anonb5dc68880502()
882 const shape_inference::ShapeAndType& s1 = (*handle_data_1)[i]; in __anonb5dc68880502()
883 const shape_inference::ShapeAndType& s2 = (*handle_data_2)[i]; in __anonb5dc68880502()
|
| D | function_ops.cc | 58 0, std::vector<shape_inference::ShapeAndType>{ in __anona02766840102()
|
| D | state_ops.cc | 285 shape_inference::ShapeAndType shape_and_type = (*handle_data)[0]; in __anon5d8c585f0502()
|
| D | functional_ops.cc | 41 const std::vector<shape_inference::ShapeAndType>* handle_type = in __anon4a8e12910102()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.h | 164 const std::vector<shape_inference::ShapeAndType>& shapes_and_types,
165 std::vector<shape_inference::ShapeAndType>* queue_shapes_and_types);
|
| D | graph_properties.cc | 47 using shape_inference::ShapeAndType;
1088 bool EquivalentShapesAndTypes(const std::vector<ShapeAndType>& st1, in EquivalentShapesAndTypes()
1089 const std::vector<ShapeAndType>& st2) const { in EquivalentShapesAndTypes()
1094 const ShapeAndType& s1 = st1[i]; in EquivalentShapesAndTypes()
1095 const ShapeAndType& s2 = st2[i]; in EquivalentShapesAndTypes()
1165 std::vector<std::unique_ptr<std::vector<ShapeAndType>>> in AddNode()
1878 const std::vector<ShapeAndType>& shapes_and_types, in RelaxEnqueueShapesAndMergeTypes()
1879 std::vector<ShapeAndType>* queue_shapes_and_types) { in RelaxEnqueueShapesAndMergeTypes()
1886 const ShapeAndType& a = shapes_and_types[i]; in RelaxEnqueueShapesAndMergeTypes()
1887 ShapeAndType& b = (*queue_shapes_and_types)[i]; in RelaxEnqueueShapesAndMergeTypes()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 39 using shape_inference::ShapeAndType;
226 std::vector<ShapeAndType>(*in_v)); in AddNode()
347 std::vector<ShapeAndType> existing_inputs; in UpdateNode()
348 const std::vector<ShapeAndType>* inputs = in UpdateNode()
783 InferenceContext* c, const std::vector<ShapeAndType>& existing, in IsUpdatedShapesOrTypes()
784 const std::vector<ShapeAndType>& updated) { in IsUpdatedShapesOrTypes()
|
| D | shape_refiner.h | 169 const std::vector<shape_inference::ShapeAndType>& existing,
170 const std::vector<shape_inference::ShapeAndType>& updated);
|
| D | shape_refiner_test.cc | 57 const std::vector<shape_inference::ShapeAndType>& existing, in IsUpdatedShapesOrTypes()
58 const std::vector<shape_inference::ShapeAndType>& updated) { in IsUpdatedShapesOrTypes()
1289 std::vector<shape_inference::ShapeAndType> t0{ in TEST_F()
1294 std::vector<shape_inference::ShapeAndType> t1{ in TEST_F()
1299 std::vector<shape_inference::ShapeAndType> t2{ in TEST_F()
1304 std::vector<shape_inference::ShapeAndType> t3{ in TEST_F()
1342 0, std::vector<shape_inference::ShapeAndType>{{shp, DT_FLOAT}}); in TEST_F()
1353 0, std::vector<shape_inference::ShapeAndType>{{shp, DT_FLOAT}}); in TEST_F()
1364 0, std::vector<shape_inference::ShapeAndType>{{shp, DT_FLOAT}}); in TEST_F()
1376 0, std::vector<shape_inference::ShapeAndType>{{shp2, DT_FLOAT}}); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | aot_only_var_handle_op.cc | 76 0, std::vector<shape_inference::ShapeAndType>{{s, t}}); in __anon1a8299070202()
|
| /external/tensorflow/tensorflow/c/ |
| D | python_api.cc | 160 std::vector<tensorflow::shape_inference::ShapeAndType> shapes_and_types; in SetHandleShapeAndType()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | shape_inference.cc | 94 0, std::vector<shape_inference::ShapeAndType>{ in PropagateShapes()
|