Home
last modified time | relevance | path

Searched refs:ShapeAndType (Results 1 – 25 of 37) sorted by relevance

12

/external/tensorflow/tensorflow/core/framework/
Dshape_inference.h133 struct ShapeAndType { struct
134 ShapeAndType() {} in ShapeAndType() argument
135 ShapeAndType(ShapeHandle s, DataType t) : shape(s), dtype(t) {} in ShapeAndType() function
136 ShapeAndType(ShapeHandle s, DataType t, SpecializedType specialized_t) in ShapeAndType() function
174 std::vector<std::unique_ptr<std::vector<ShapeAndType>>>
370 std::string DebugString(const ShapeAndType& shape_and_type);
371 std::string DebugString(gtl::ArraySlice<ShapeAndType> shape_and_types);
570 const std::vector<ShapeAndType>& shapes_and_types) TF_MUST_USE_RESULT;
575 const std::vector<ShapeAndType>& shapes_and_types) TF_MUST_USE_RESULT;
588 const std::vector<ShapeAndType>& shapes_and_types) TF_MUST_USE_RESULT;
[all …]
Dshape_inference.cc63 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()
195 std::vector<std::unique_ptr<std::vector<ShapeAndType>>> input_handle_data) { in PostInputInit()
294 string InferenceContext::DebugString(const ShapeAndType& shape_and_type) { in DebugString()
300 gtl::ArraySlice<ShapeAndType> shape_and_types) { in DebugString()
302 for (const ShapeAndType& s : shape_and_types) { in DebugString()
1171 const std::vector<ShapeAndType>& shapes_and_types, in MergeHandleShapesAndTypes()
1172 std::vector<ShapeAndType>* to_update) { in MergeHandleShapesAndTypes()
1176 std::vector<ShapeAndType> new_values(shapes_and_types.size()); in MergeHandleShapesAndTypes()
[all …]
Dshape_inference_testutil.h35 typedef std::pair<string, DataType> ShapeAndType; typedef
40 std::vector<std::vector<ShapeAndType>*>
Dshape_inference_testutil.cc45 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()
Dshape_inference_test.cc1622 [&](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()
/external/tensorflow/tensorflow/core/ops/
Dsparse_csr_matrix_ops.cc26 using shape_inference::ShapeAndType;
30 ShapeAndType* shape_and_type) { in GetVariantInput()
83 {ShapeAndType{dense_shape, dtype}}); in __anon59a9d6c30102()
94 ShapeAndType sparse_matrix_shape_and_type; in __anon59a9d6c30202()
146 {ShapeAndType{dense_shape, dtype}}); in __anon59a9d6c30302()
156 ShapeAndType sparse_matrix_shape_and_type; in __anon59a9d6c30402()
175 ShapeAndType sparse_matrix_shape_and_type; in __anon59a9d6c30502()
203 ShapeAndType sparse_matrix_shape_and_type; in __anon59a9d6c30602()
233 ShapeAndType sparse_matrix_shape_and_type; in __anon59a9d6c30702()
304 ShapeAndType sparse_matrix_shape_and_type; in __anon59a9d6c30802()
[all …]
Dsparse_csr_matrix_ops_test.cc45 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 …]
Dlist_ops.cc28 const std::vector<shape_inference::ShapeAndType>& shapes_and_types, in VerifyHandleData()
36 const shape_inference::ShapeAndType& list_shape_type = shapes_and_types[0]; in VerifyHandleData()
47 const std::vector<shape_inference::ShapeAndType>* handle_data) { in IsValidTensorListHandleData()
53 const std::vector<shape_inference::ShapeAndType>& shapes_and_types) { in GetElementShapeFromHandleData()
71 0, std::vector<shape_inference::ShapeAndType>{ in __anonabedecb60202()
93 const shape_inference::ShapeAndType& list_shape_type = in __anonabedecb60302()
108 0, std::vector<shape_inference::ShapeAndType>{ in __anonabedecb60302()
140 const shape_inference::ShapeAndType& list_shape_type = in __anonabedecb60402()
155 0, std::vector<shape_inference::ShapeAndType>{ in __anonabedecb60402()
181 const shape_inference::ShapeAndType& list_shape_type = in __anonabedecb60502()
[all …]
Dresource_variable_ops.cc25 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()
95 std::vector<ShapeAndType>{{s, t}}); in __anon50d007a70202()
128 i, std::vector<ShapeAndType>{{s, dtypes[i]}}); in __anon50d007a70302()
170 std::vector<ShapeAndType> handle_shape_and_type; in CreateAssignShapeFn()
246 std::vector<ShapeAndType> handle_shape_and_type; in __anon50d007a70402()
287 std::vector<ShapeAndType> variant_shape_and_type; in __anon50d007a70402()
307 std::vector<ShapeAndType> handle_shape_and_type; in ResourceScatterUpdateShape()
Dlookup_ops.cc26 using shape_inference::ShapeAndType;
91 const ShapeAndType& key_shape_and_type, in ValidateTableType()
93 const ShapeAndType& value_shape_and_type, in ValidateTableType()
119 ShapeAndType* output_shape_and_type) { in ValidateTableResourceHandle()
125 const ShapeAndType& key_shape_and_type = (*handle_data)[0]; in ValidateTableResourceHandle()
126 const ShapeAndType& value_shape_and_type = (*handle_data)[1]; in ValidateTableResourceHandle()
172 ShapeAndType value_shape_and_type; in __anon18b155f40302()
271 const ShapeAndType& key_shape_and_type = (*handle_data)[0]; in __anon18b155f40802()
272 const ShapeAndType& value_shape_and_type = (*handle_data)[1]; in __anon18b155f40802()
331 0, std::vector<ShapeAndType>{{key_s, key_t}, {value, value_t}}); in MutableHashTableShape()
Dlookup_ops_test.cc35 std::vector<std::vector<ShapeInferenceTestOp::ShapeAndType>> types; in TEST()
65 std::vector<std::vector<ShapeInferenceTestOp::ShapeAndType>> types; in TEST()
Ddata_flow_ops_test.cc205 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()
Dtpu_replication_ops.cc63 const std::vector<shape_inference::ShapeAndType>* shapes_and_types = in __anonbea61acc0102()
Dfunction_ops.cc83 0, std::vector<shape_inference::ShapeAndType>{{shape_handle, dtype}}); in __anon5bd79b600102()
Dstate_ops_test.cc80 std::vector<ShapeInferenceTestOp::ShapeAndType> shapes_and_types; in TEST()
Dmath_ops.cc52 std::vector<shape_inference::ShapeAndType> cur_shapes_and_types; in __anon725c8f640102()
783 std::vector<shape_inference::ShapeAndType> merged_handle_data(size); in __anon725c8f640402()
791 const shape_inference::ShapeAndType& s1 = (*handle_data_1)[i]; in __anon725c8f640402()
792 const shape_inference::ShapeAndType& s2 = (*handle_data_2)[i]; in __anon725c8f640402()
872 std::vector<shape_inference::ShapeAndType> merged_handle_data(size); in __anon725c8f640502()
880 const shape_inference::ShapeAndType& s1 = (*handle_data_1)[i]; in __anon725c8f640502()
881 const shape_inference::ShapeAndType& s2 = (*handle_data_2)[i]; in __anon725c8f640502()
Dfunctional_ops.cc41 const std::vector<shape_inference::ShapeAndType>* handle_type = in __anonb31eee6d0102()
/external/tensorflow/tensorflow/core/tpu/ops/
Dtpu_partitioned_input_op.cc64 const std::vector<shape_inference::ShapeAndType>* shapes_and_types = in __anon2b8a3a590102()
93 std::vector<shape_inference::ShapeAndType> output_shapes_and_types; in __anon2b8a3a590102()
94 output_shapes_and_types.push_back(shape_inference::ShapeAndType( in __anon2b8a3a590102()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.h165 const std::vector<shape_inference::ShapeAndType>& shapes_and_types,
166 std::vector<shape_inference::ShapeAndType>* queue_shapes_and_types);
Dgraph_properties.cc48 using shape_inference::ShapeAndType;
1251 bool EquivalentShapesAndTypes(const std::vector<ShapeAndType>& st1, in EquivalentShapesAndTypes()
1252 const std::vector<ShapeAndType>& st2) const { in EquivalentShapesAndTypes()
1257 const ShapeAndType& s1 = st1[i]; in EquivalentShapesAndTypes()
1258 const ShapeAndType& s2 = st2[i]; in EquivalentShapesAndTypes()
1334 std::vector<std::unique_ptr<std::vector<ShapeAndType>>> in AddNode()
2193 const std::vector<ShapeAndType>& shapes_and_types, in RelaxEnqueueShapesAndMergeTypes()
2194 std::vector<ShapeAndType>* queue_shapes_and_types) { in RelaxEnqueueShapesAndMergeTypes()
2201 const ShapeAndType& a = shapes_and_types[i]; in RelaxEnqueueShapesAndMergeTypes()
2202 ShapeAndType& b = (*queue_shapes_and_types)[i]; in RelaxEnqueueShapesAndMergeTypes()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc39 using shape_inference::ShapeAndType;
231 std::vector<ShapeAndType>(*in_v)); in AddNodeInternal()
352 std::vector<ShapeAndType> existing_inputs; in UpdateNode()
353 const std::vector<ShapeAndType>* inputs = in UpdateNode()
792 InferenceContext* c, const std::vector<ShapeAndType>& existing, in IsUpdatedShapesOrTypes()
793 const std::vector<ShapeAndType>& updated) { in IsUpdatedShapesOrTypes()
Dshape_refiner.h170 const std::vector<shape_inference::ShapeAndType>& existing,
171 const std::vector<shape_inference::ShapeAndType>& updated);
Dshape_refiner_test.cc57 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/
Daot_only_var_handle_op.cc76 0, std::vector<shape_inference::ShapeAndType>{{s, t}}); in __anon5de01ee30202()
/external/tensorflow/tensorflow/c/
Dpython_api.cc125 std::vector<tensorflow::shape_inference::ShapeAndType> shapes_and_types; in SetHandleShapeAndType()

12