Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Dshape_inference.h140 struct ShapeAndType { struct
141 ShapeAndType() {} in ShapeAndType() argument
142 ShapeAndType(ShapeHandle s, DataType t) : shape(s), dtype(t) {} in ShapeAndType() function
178 std::vector<std::unique_ptr<std::vector<ShapeAndType>>>
384 string DebugString(const ShapeAndType& shape_and_type);
385 string DebugString(gtl::ArraySlice<ShapeAndType> shape_and_types);
580 const std::vector<ShapeAndType>& shapes_and_types) TF_MUST_USE_RESULT;
585 const std::vector<ShapeAndType>& shapes_and_types) TF_MUST_USE_RESULT;
598 const std::vector<ShapeAndType>& shapes_and_types) TF_MUST_USE_RESULT;
603 const std::vector<ShapeAndType>& shapes_and_types) TF_MUST_USE_RESULT;
[all …]
Dshape_inference.cc64 std::vector<std::unique_ptr<std::vector<ShapeAndType>>> handle_data( in InferenceContext()
71 handle_data[i].reset(new std::vector<ShapeAndType>(v->size())); in InferenceContext()
118 std::vector<std::unique_ptr<std::vector<ShapeAndType>>> handle_data( in InferenceContext()
125 handle_data[i].reset(new std::vector<ShapeAndType>(v->size())); in InferenceContext()
145 std::vector<std::unique_ptr<std::vector<ShapeAndType>>> in InferenceContext()
252 std::vector<std::unique_ptr<std::vector<ShapeAndType>>> input_handle_data) { in PostInputInit()
352 string InferenceContext::DebugString(const ShapeAndType& shape_and_type) { in DebugString()
358 gtl::ArraySlice<ShapeAndType> shape_and_types) { in DebugString()
360 for (const ShapeAndType& s : shape_and_types) { in DebugString()
1196 const std::vector<ShapeAndType>& shapes_and_types, in MergeHandleShapesAndTypes()
[all …]
Dshape_inference_testutil.h34 typedef std::pair<string, DataType> ShapeAndType; typedef
39 std::vector<std::vector<ShapeAndType>*>
Dshape_inference_testutil.cc46 std::vector<std::unique_ptr<std::vector<shape_inference::ShapeAndType>>> in InferShapes()
52 std::unique_ptr<std::vector<ShapeAndType>> v( in InferShapes()
53 new std::vector<ShapeAndType>()); in InferShapes()
Dshape_inference_test.cc1623 [&](int idx, const std::vector<ShapeAndType>& shapes_and_types) { in TestMergeHandles()
1635 std::vector<ShapeAndType> t{{make_shape({1, 2, 3}), DT_FLOAT}, in TestMergeHandles()
1640 std::vector<ShapeAndType> v = *get_shapes_and_types_from_context(0); in TestMergeHandles()
1650 0, std::vector<ShapeAndType>{{make_shape({1, 2, 3}), DT_FLOAT}})); in TestMergeHandles()
1734 [&](int idx, const std::vector<ShapeAndType>& shapes_and_types) { in TestRelaxHandles()
1746 std::vector<ShapeAndType> t{{make_shape({1, 2, 3}), DT_FLOAT}, in TestRelaxHandles()
1751 std::vector<ShapeAndType> v = *get_shapes_and_types_from_context(0); in TestRelaxHandles()
1761 0, std::vector<ShapeAndType>{{make_shape({1, 2, 3}), DT_FLOAT}})); in TestRelaxHandles()
/external/tensorflow/tensorflow/core/ops/
Dlist_ops.cc27 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 __anonebbd57d80202()
87 const shape_inference::ShapeAndType& list_shape_type = in __anonebbd57d80302()
102 0, std::vector<shape_inference::ShapeAndType>{ in __anonebbd57d80302()
134 const shape_inference::ShapeAndType& list_shape_type = in __anonebbd57d80402()
149 0, std::vector<shape_inference::ShapeAndType>{ in __anonebbd57d80402()
175 const shape_inference::ShapeAndType& list_shape_type = in __anonebbd57d80502()
212 const shape_inference::ShapeAndType& list_shape_type = in __anonebbd57d80602()
[all …]
Dresource_variable_ops.cc25 using ::tensorflow::shape_inference::ShapeAndType;
33 InferenceContext* c, std::vector<ShapeAndType>* shape_and_type) { in ValidateVariableResourceHandle()
53 std::vector<ShapeAndType> shape_and_type; in ReadVariableShapeFn()
57 std::vector<ShapeAndType> variant_shape_and_type; in ReadVariableShapeFn()
75 ShapeAndType shape_and_type; in ReadVariablesShapeFn()
113 std::vector<ShapeAndType>{{s, t}}); in __anon3f9cfa090202()
146 i, std::vector<ShapeAndType>{{s, dtypes[i]}}); in __anon3f9cfa090302()
188 std::vector<ShapeAndType> handle_shape_and_type; in CreateAssignShapeFn()
263 std::vector<ShapeAndType> handle_shape_and_type; in __anon3f9cfa090402()
304 std::vector<ShapeAndType> variant_shape_and_type; in __anon3f9cfa090402()
[all …]
Dlookup_ops.cc26 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 __anon8a0811960302()
272 ShapeAndType value_shape_and_type; in __anon8a0811960802()
331 0, std::vector<ShapeAndType>{{key_s, key_t}, {value, value_t}}); in MutableHashTableShape()
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.cc56 const std::vector<shape_inference::ShapeAndType>* shapes_and_types = in __anon576f0cae0102()
Dstate_ops.cc285 shape_inference::ShapeAndType shape_and_type = (*handle_data)[0]; in __anone3ce2a1d0502()
Dfunctional_ops.cc41 const std::vector<shape_inference::ShapeAndType>* handle_type = in __anon22ceab0f0102()
Dmath_ops.cc52 std::vector<shape_inference::ShapeAndType> cur_shapes_and_types; in __anonb22bfa860102()
746 std::vector<shape_inference::ShapeAndType> merged_handle_data(size); in __anonb22bfa860502()
754 const shape_inference::ShapeAndType& s1 = (*handle_data_1)[i]; in __anonb22bfa860502()
755 const shape_inference::ShapeAndType& s2 = (*handle_data_2)[i]; in __anonb22bfa860502()
Ddata_flow_ops.cc587 0, std::vector<shape_inference::ShapeAndType>{{s, t}}); in __anon4a7ac08d1102()
676 shape_inference::ShapeAndType shape_and_type = (*handle_data)[0]; in __anon4a7ac08d1402()
Darray_ops.cc355 0, std::vector<shape_inference::ShapeAndType>({})); in __anon7c94107b0302()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.h135 const std::vector<shape_inference::ShapeAndType>& shapes_and_types,
136 std::vector<shape_inference::ShapeAndType>* queue_shapes_and_types);
Dgraph_properties.cc46 using shape_inference::ShapeAndType;
1017 bool EquivalentShapesAndTypes(const std::vector<ShapeAndType>& st1, in EquivalentShapesAndTypes()
1018 const std::vector<ShapeAndType>& st2) const { in EquivalentShapesAndTypes()
1023 const ShapeAndType& s1 = st1[i]; in EquivalentShapesAndTypes()
1024 const ShapeAndType& s2 = st2[i]; in EquivalentShapesAndTypes()
1085 std::vector<std::unique_ptr<std::vector<ShapeAndType>>> in AddNode()
1760 const std::vector<ShapeAndType>& shapes_and_types, in RelaxEnqueueShapesAndMergeTypes()
1761 std::vector<ShapeAndType>* queue_shapes_and_types) { in RelaxEnqueueShapesAndMergeTypes()
1768 const ShapeAndType& a = shapes_and_types[i]; in RelaxEnqueueShapesAndMergeTypes()
1769 ShapeAndType& b = (*queue_shapes_and_types)[i]; in RelaxEnqueueShapesAndMergeTypes()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc39 using shape_inference::ShapeAndType;
217 std::vector<std::unique_ptr<std::vector<ShapeAndType>>> in AddNode()
240 new std::vector<ShapeAndType>(*in_v)); in AddNode()
374 std::vector<ShapeAndType> existing_inputs; in UpdateNode()
375 const std::vector<ShapeAndType>* inputs = in UpdateNode()
749 InferenceContext* c, const std::vector<ShapeAndType>& existing, in IsUpdatedShapesOrTypes()
750 const std::vector<ShapeAndType>& updated) { in IsUpdatedShapesOrTypes()
Dshape_refiner.h199 const std::vector<shape_inference::ShapeAndType>& existing,
200 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()
1304 std::vector<shape_inference::ShapeAndType> t0{ in TEST_F()
1309 std::vector<shape_inference::ShapeAndType> t1{ in TEST_F()
1314 std::vector<shape_inference::ShapeAndType> t2{ in TEST_F()
1319 std::vector<shape_inference::ShapeAndType> t3{ in TEST_F()
1357 0, std::vector<shape_inference::ShapeAndType>{{shp, DT_FLOAT}}); in TEST_F()
1368 0, std::vector<shape_inference::ShapeAndType>{{shp, DT_FLOAT}}); in TEST_F()
1379 0, std::vector<shape_inference::ShapeAndType>{{shp, DT_FLOAT}}); in TEST_F()
1391 0, std::vector<shape_inference::ShapeAndType>{{shp2, DT_FLOAT}}); in TEST_F()
/external/tensorflow/tensorflow/contrib/framework/ops/
Dvariable_ops.cc57 0, std::vector<shape_inference::ShapeAndType>{{s, t}}); in __anonf00e47a70202()
/external/tensorflow/tensorflow/compiler/jit/
Dshape_inference.cc83 0, std::vector<shape_inference::ShapeAndType>{ in PropagateShapes()
/external/tensorflow/tensorflow/c/
Dpython_api.cc161 std::vector<tensorflow::shape_inference::ShapeAndType> shapes_and_types; in SetHandleShapeAndType()
Dc_api.cc739 std::vector<tensorflow::shape_inference::ShapeAndType>( in TF_GraphSetOutputHandleShapesAndTypes()
744 shape_and_type_vec[i] = tensorflow::shape_inference::ShapeAndType( in TF_GraphSetOutputHandleShapesAndTypes()