Searched defs:shape_data (Results 1 – 9 of 9) sorted by relevance
49 std::initializer_list<int> shape_data, in ReshapeOpModel()82 std::initializer_list<int> shape_data) { in BuildWithHardcodedShape()96 std::initializer_list<int> shape_data) { in BuildWithTensorShape()116 std::initializer_list<int> shape_data) { in BuildWithConstantTensorShape()
41 const auto* shape_data = GetTensorData<IndicesT>(shape); in ResizeOutputTensor() local53 const IndicesT* shape_data) { in CheckShapes()
373 const Tensor* shape_data = c->input_tensor(0); in __anon45ba5f3d0602() local392 const Tensor* shape_data = c->input_tensor(0); in __anon45ba5f3d0702() local413 const Tensor* shape_data = c->input_tensor(0); in __anon45ba5f3d0802() local467 Node* shape_data; in TEST_F() local501 Node* shape_data; in TEST_F() local534 Node* shape_data; in TEST_F() local560 Node* shape_data; in TEST_F() local586 Node* shape_data; in TEST_F() local617 Node* shape_data; in TEST_F() local647 Node* shape_data; in TEST_F() local[all …]
116 const int* shape_dims_data, const int32_t* shape_data, in TestReshape()135 const int* shape_dims_data, const int32_t* shape_data, in TestReshapeQuantized()345 const int32_t shape_data[] = {0}; in TF_LITE_MICRO_TEST() local
840 const std::vector<int32> shape_data = { in ConvertSoftmaxOperator() local882 const std::vector<int32> shape_data = { in ConvertLogSoftmaxOperator() local1065 const auto& shape_data = shape_array.GetBuffer<ArrayDataType::kInt32>().data; in ConvertTensorFlowReshapeOperator() local
210 std::vector<int64_t> shape_data = {1, 1, 1, 1}; in ExpandTo4DForConvImpl() local
428 std::vector<int32> shape_data = in ProcessTensorFlowReshapeOperator() local
2642 const int32_t* shape_data = in VisitResizeBilinearNode() local
1328 SmallVector<int64_t, 4> shape_data; in fold() local