Searched refs:is_shape_tensor_ (Results 1 – 2 of 2) sorted by relevance
124 FixShapeValueProfile(&prof, is_shape_tensor_); in InitProfiles()190 is_shape_tensor_.resize(n_inputs, false); in SetShapeTensorMask()194 is_shape_tensor_[i] = input->isShapeTensor(); in SetShapeTensorMask()195 if (is_shape_tensor_[i]) { in SetShapeTensorMask()201 absl::c_any_of(is_shape_tensor_, [](bool b) { return b; }); in SetShapeTensorMask()209 is_shape_tensor_.resize(input_partial_shapes.size(), false); in SetShapeTensorMask()211 is_shape_tensor_[i] = IsTrtShapeTensorCompatible(input_partial_shapes[i]); in SetShapeTensorMask()212 if (is_shape_tensor_[i]) { in SetShapeTensorMask()217 absl::c_any_of(is_shape_tensor_, [](bool b) { return b; }); in SetShapeTensorMask()
196 std::vector<bool> is_shape_tensor_; variable