Searched defs:shape_data (Results 1 – 11 of 11) sorted by relevance
375 const Tensor* shape_data = c->input_tensor(0); in __anonffbadc8c0602() local394 const Tensor* shape_data = c->input_tensor(0); in __anonffbadc8c0702() local415 const Tensor* shape_data = c->input_tensor(0); in __anonffbadc8c0802() local469 Node* shape_data; in TEST_F() local503 Node* shape_data; in TEST_F() local536 Node* shape_data; in TEST_F() local562 Node* shape_data; in TEST_F() local588 Node* shape_data; in TEST_F() local619 Node* shape_data; in TEST_F() local649 Node* shape_data; in TEST_F() local[all …]
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()
87 auto shape_data = GetTensorData<int32_t>(shape_tensor); in ResizeOutputAndTemporaryTensors() local
312 SmallVector<int64_t, 4> shape_data; in matchAndRewrite() local
274 std::vector<int64_t> shape_data = {1, 1, 1, 1}; in ExpandTo4DForConvImpl() local
845 const std::vector<int32> shape_data = { in ConvertSoftmaxOperator() local887 const std::vector<int32> shape_data = { in ConvertLogSoftmaxOperator() local1070 const auto& shape_data = shape_array.GetBuffer<ArrayDataType::kInt32>().data; in ConvertTensorFlowReshapeOperator() local
249 const DimsAdapter& shape_data) noexcept { in ConstantShape()
428 std::vector<int32> shape_data = in ProcessTensorFlowReshapeOperator() local
1734 SmallVector<int64_t, 4> shape_data; in fold() local
4244 const int32_t* shape_data = in VisitResizeBilinearNode() local